Location: Hodgkin & Huxley (1952) model @ ab2962b87966 / sed-ml / originalFig21_plot.py

Author:
WeiweiAi <wai484@aucklanduni.ac.nz>
Date:
2022-03-31 10:48:40+13:00
Desc:
Add open channel IV of HH models; Add simulation and plot python scripts
Permanent Source URI:
https://models.cellml.org/workspace/64f/rawfile/ab2962b8796666ad9938a2577611c954f006b5fd/sed-ml/originalFig21_plot.py

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

# The prefix of the saved output file name 
prefilename = 'simFig21'
# Figure name
prefig = 'Fig21'
figfile = 'original%s.png' % prefig
# Set figure dimension (width, height) in inches.
fw, fh = 6, 6
fig = plt.figure(figsize=(fw,fh))
# This gives list with the colors from the cycle, which you can use to iterate over.
cycle = plt.rcParams['axes.prop_cycle'].by_key()['color']
# Set subplots
subpRow, subpCol = 1, 1
ax = fig.add_subplot(111)
lfontsize, labelfontsize = 12, 12 # legend, label fontsize
# For read data from the simulation output files
V_initial = [10, -7, -6, -5, -2]
x_name = 'outputs/time'
y_name ='outputs/minus_V'
rol=int(10*100)
for i, iV_initial in enumerate(V_initial):
    filename ='%s_(%d)mV.csv' % (prefilename, iV_initial)      
    data = pd.read_csv(filename)
    x_data = data[x_name][0:rol]   
    y_data = data[y_name][0:rol]
    T_data = data['parameters/T']
    ax.plot(x_data, y_data, color=cycle[i], label = 'CellML @ %d mV' % (iV_initial) )

    filename = 'fig21_(%d).csv' % iV_initial
    odata = pd.read_csv(filename)
    ox_data = odata['x']   
    oy_data = odata['Curve1']
    ax.plot(ox_data, oy_data, '.', color=cycle[i], label = 'HH @ %d mV' % (iV_initial ) )

    ax.tick_params(direction='in', axis='both')    
    ax.legend(loc = 'upper right', fontsize=lfontsize, frameon=False)
    ax.set_xlabel ('time (ms)', fontsize= labelfontsize)
    ax.set_ylabel ('-V (mV)', fontsize= labelfontsize)

plt.ylim([-11, 20])
ax.set_title('%s in the primary publication' % (prefig))
plt.grid(True,linestyle='-.')
plt.axhline(y=0.0, color="black", linestyle="--")
plt.savefig(figfile)        
plt.show()