INNER CODE UNIT · Python
polarization_plot2
ECSIM/pem-dataset1 · Notebooks/functions.py:255
def polarization_plot2(data, set_index):
"""
Polarization plot function 2.
:param data: input data
:type data: numpy array
:param set_index: activation set index
:type set_index: int or list
:return: None
"""
data_I = []
data_V = []
data_P = []
if isinstance(set_index,int):
set_list = [set_index]
else:
set_list = sorted(set_index)
for i in set_list: