- Author:
- WeiweiAi <wai484@aucklanduni.ac.nz>
- Date:
- 2021-06-09 17:59:55+12:00
- Desc:
- Update figure titles and documentation;
- Permanent Source URI:
- https://models.cellml.org/workspace/692/rawfile/4de8d2594e8e71c62b4f118f49fde8a277df70d7/Simulation/Fig13_plot.py
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
# The prefix of the saved output file name
prefilename = 'simFig13_'
# Figure name
prefig = 'Fig13'
figfile = 'sim%s.png' % prefig
# Set figure dimension (width, height) in inches.
fw, fh = 12, 10
fig = plt.figure(figsize=(fw,fh))
fig.tight_layout()
# Set subplots
subpRow, subpCol = 3, 2
ax, lns = {}, {}
lfontsize, labelfontsize = 12, 12 # 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
gnames = ['g_Kv', 'g_BK', 'g_Na', 'g_CaL']
fileindex = ['a', 'b', 'c', 'd', 'e']
labels= ['CellML model control','CellML model +50%','CellML model -50%',]
plotindex = [1, 2, 3, 4, 5, 6 ]
x_name = 'time'
y_name = ['V', 'Cai']
for i, ifile in enumerate(fileindex):
for n in range(3):
if n==0:
filename='%s%d.csv' % (prefilename,n)
ax[i] = fig.add_subplot(subpRow, subpCol, plotindex[i])
ofilename ='fig13%s%d.csv' % (fileindex[i],n)
odata = pd.read_csv(ofilename)
ox_data = odata['x']
oy_data = odata['Curve1']
ax[i].plot(ox_data, oy_data, '.', color=cycle[n])
else:
if i < 4:
filename='%s%s%d.csv' % (prefilename,fileindex[i],n)
else:
filename='%s%s%d.csv' % (prefilename,fileindex[3],n)
if i in [0, 4]:
ofilename ='fig13%s%d.csv' % (fileindex[i],n)
odata = pd.read_csv(ofilename)
ox_data = odata['x']
oy_data = odata['Curve1']
ax[i].plot(ox_data, oy_data, '.', color=cycle[n])
data = pd.read_csv(filename)
x_data = data[x_name]/1000
if i < 4:
y_data = data[y_name[0]]
else:
y_data = data[y_name[1]]*10**6
ax[i].plot(x_data, y_data, color=cycle[n], label = labels[n] )
ax[i].tick_params(direction='in', axis='both')
if i < 4:
ax[i].set_ylabel ('Voltage (mV)', fontsize= labelfontsize)
ax[i].set_title(gnames[i])
else:
ax[i].set_ylabel ('[Ca]$_i$ (nM)', fontsize= labelfontsize)
ax[i].set_title(gnames[3])
ax[i].legend(bbox_to_anchor=(1.04,1), loc="upper left", fontsize=lfontsize, frameon=False)
if plotindex[i] in [5, 4]:
ax[i].set_xlabel ('time(ms)', fontsize= labelfontsize)
else:
ax[i].set_xticklabels([])
ax[i].set_title('%s in the paper' % (prefig))
plt.savefig(figfile)
plt.show()
