如何在python中以不同颜色在同一图形中绘制10条轨迹?

如何在python中以不同颜色在同一图形中绘制10条轨迹?,python,numpy,plot,Python,Numpy,Plot,我需要在python中用不同的颜色绘制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'

我需要在python中用不同的颜色绘制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()
根据你的说法,我必须把它们都放在同一个文件里吗?为了策划他们


提前谢谢

不,您不必将所有内容都放在同一个文件中。您可以简单地在文件列表上循环并打印到相同的
轴上。对于颜色,如果您只需为
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()