Как построить 10 следов на одном рисунке с другим цветом в python? .

Question

мне нужно построить 10 трасс с различным цветом в питона, каждый след в другом файле с тем же расширением .numpy, я имею в виду, что это у меня есть 10 файлов:

trace1 
trace2 
trace3 
trace4 
trace5 
trace6 
trace7 
trace8 
trace9 
trace10 

Это мой код для построения всего одного следа:

import matplotlib.pyplot as plt 
import numpy as np 
dataArray= np.load(r'/home/user/trace1.npy') 
print(dataArray) 
plt.plot(dataArray.T) 
plt.show() 

Согласно вам, я должен положить их все в одном файле? Чтобы их нарисовать?

Заранее спасибо.

Answer 1

Нет, вам не нужно класть все в один и тот же файл. Вы можете просто перебрать список файлов и занести в один и тот же номер axes. Для цвета это самый простой, если вы просто захватите цвет для colormap. Вот небольшой пример:

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

# Read in list of files. You might want to look into os.listdir() 
traces=[list of filepaths to your .npy files] 

# Create figure 
fig=plt.figure() 
fig.show() 
ax=fig.add_subplot(111) 

# Grab colormap 
cmap = matplotlib.cm.get_cmap('jet') 

# Loop through traces and plot them 
for j,trace in enumerate(traces): 

    # Load file 
    dataArray= np.load(trace) 

    # Grab color 
    c=cmap(float(j)/len(traces)) 

    # Plot 
    ax.plot(dataArray.T,color=c) 

plt.show()