diff --git a/mbtrack2/impedance/impedance_model.py b/mbtrack2/impedance/impedance_model.py index bc3da3c0f8f7b3bb2794e580668c1efdc1783239..46f649c53c142d00493ae2c2e1f4d3b463a6cb68 100644 --- a/mbtrack2/impedance/impedance_model.py +++ b/mbtrack2/impedance/impedance_model.py @@ -5,7 +5,7 @@ Module where the ImpedanceModel class is defined. import pandas as pd import numpy as np import matplotlib.pyplot as plt -from matplotlib import colormaps +#from matplotlib import colormaps import pickle from copy import deepcopy from scipy.integrate import trapz @@ -338,14 +338,15 @@ class ImpedanceModel(): legend.append("Spectral density for sigma = " + str(sigma) + " s") # Main plot - colorblind = colormaps['tab10'].colors + #colorblind = colormaps['tab10'].colors for index in sorted_index: attr = attr_list[index] # Set all impedances with common indexes using + zero_impedance try: sum_imp = getattr(getattr(self, attr), Z_type) + zero_impedance ax.fill_between(sum_imp.data.index*1e-9, total_imp, - total_imp + sum_imp.data[component]*scale[leg], edgecolor=colorblind[index % 10], color=colorblind[index % 10]) + total_imp + sum_imp.data[component]*scale[leg]) + #total_imp + sum_imp.data[component]*scale[leg], edgecolor=colorblind[index % 10], color=colorblind[index % 10]) total_imp += sum_imp.data[component]*scale[leg] if attr[:4] == "sum_": legend.append(attr[4:]) @@ -378,7 +379,8 @@ class ImpedanceModel(): sum_imp = getattr(getattr(self, attr), Z_type) + zero_impedance in_ax.fill_between(sum_imp.data.index*1e-3, total_imp, - total_imp + sum_imp.data[component]*1e-9, edgecolor=colorblind[index % 10], color=colorblind[index % 10]) + total_imp + sum_imp.data[component]*1e-9) + # total_imp + sum_imp.data[component]*1e-9, edgecolor=colorblind[index % 10], color=colorblind[index % 10]) total_imp += sum_imp.data[component]*1e-9 except AttributeError: pass