[英]plot in the same figure multiple lines from different data files with python
[英]How to plot 10 traces in same figure with different color in python?
我需要在 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()
根據您的說法,我必須將它們全部放在同一個文件中嗎? 為了圖他們?
提前致謝。
不,您不必將所有內容都放在同一個文件中。 您可以簡單地遍歷文件列表並繪制到相同的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()
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.