如何在python中以不同颜色在同一图形中绘制10条轨迹?
我需要在python中用不同的颜色绘制10个跟踪,每个跟踪都在不同的文件中,具有相同的扩展名.numpy.,我的意思是我有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'
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()