- Author:
- WeiweiAi <wai484@aucklanduni.ac.nz>
- Date:
- 2021-11-24 17:49:12+13:00
- Desc:
- Add BG params related scripts and matrix
- Permanent Source URI:
- http://models.cellml.org/workspace/64f/rawfile/bd02dedb491faf9960ccd23496c19ccd3e339db1/sed-ml/IstGating_plot.py
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
# The prefix of the saved output file name
prefilename = 'Ist_20ms'
suf=['-10']
# Figure name
prefig = 'IstGating'
figfile = '%s.png' % prefig
# Set figure dimension (width, height) in inches.
fw, fh = 6, 9
fig = plt.figure(figsize=(fw,fh))
# Set subplots
subpRow, subpCol = 3, 1
lns = {}
lfontsize, labelfontsize = 10, 10 # legend, label fontsize
# 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']
# Read data from the files
plotindex = [1, 2, 3]
x_name = 'outputs/time'
y_names =[['outputs/minus_V','outputs/m','outputs/h','outputs/n',],['outputs/minus_V','outputs/gNa','outputs/gK',],
['outputs/minus_V','outputs/INa','outputs/IK','outputs/Ileak','outputs/Ii']]
yl_names =[['-V','m','h','n',],['-V','$g_{Na}$','$g_{K}$',],
['-V','$I_{Na}$','$I_{K}$','$I_{l}$','$I_{i}$']]
y_labels =[['-V (mV)','gating'], ['-V (mV)','conductance'], ['-V (mV)','Currents']]
rol=int(30*1000)
w, h = 2, len(y_labels)
ax = [[0 for x in range(w)] for y in range(h)]
for j, iV_initial in enumerate(suf):
filename ='%s%smV.csv' % (prefilename, iV_initial)
data = pd.read_csv(filename)
x_data = data[x_name][0:rol]
for i, y_name in enumerate(y_names):
for m, iy_name in enumerate(y_name):
y_data = data[iy_name][0:rol]
if m == 0:
ax[i][0] = fig.add_subplot(subpRow, subpCol, plotindex[i])
ax[i][0].plot(x_data, y_data,'-.', color=cycle[m],label='%s' % (yl_names[i][m]))
ax[i][0].set_ylabel (y_labels[i][0], fontsize= labelfontsize)
ax[i][0].tick_params(direction='in', axis='both')
ax[i][0].grid(axis='x',linestyle='-.')
elif m == 1:
ax[i][1] = ax[i][0].twinx()
lns[m]= ax[i][1].plot(x_data, y_data, color=cycle[m], label='%s' % (yl_names[i][m]))
sumlns=lns[m]
ax[i][1].set_ylabel (y_labels[i][1], fontsize= labelfontsize)
ax[i][1].tick_params(direction='in', axis='both')
else:
lns[m]= ax[i][1].plot(x_data, y_data, color=cycle[m], label='%s' % (yl_names[i][m]))
sumlns=sumlns+lns[m]
if plotindex[i] in [3]:
ax[i][0].set_xlabel ('time(ms)', fontsize= labelfontsize)
else:
ax[i][0].set_xticklabels([])
ax[i][1].legend(sumlns, yl_names[i][1:], frameon=False, loc= 'upper right')
plt.savefig(figfile)
plt.show()
