我的 result.csv 有 8 列,需要向我的折线图添加图例。我的代码是:
per_data=genfromtxt('result.csv',delimiter=',')
plt.plot(per_data)
plt.xlabel ('x stuff')
plt.ylabel ('y stuff')
plt.title('my test result')
plt.grid()
plt.show()
它给了我:
如何添加恰好是 csv 文件中标题行的图例?
最佳答案
如果您对 np.genfromtxt
使用 names=True
选项,它将在 .csv
的第一行中读取作为列名称。
例如:
import matplotlib.pyplot as plt
import numpy as np
# Make dummy csv file for this example
from io import StringIO
result_csv = StringIO(u"""
xstuff, data1, data2, data3
0, 1, 2, 3
1, 1, 3, 4
2, 2, 1, 3
3, 1, 2, 5
""")
# Read in csv. Use names=True to also store column headers
per_data=np.genfromtxt(result_csv,delimiter=',',names=True)
# Loop over columns. Here I assume you have the x-data in the first column, so skip that one
for name in per_data.dtype.names[1:]:
# Set the line's label to the column name
plt.plot(per_data['xstuff'],per_data[name],label=name)
# Add a legend
plt.legend(loc=0)
plt.xlabel ('x stuff')
plt.ylabel ('y stuff')
plt.title('my test result')
plt.grid()
plt.show()
关于python - matplotlib 将 csv 标题行的图例添加到折线图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36877918/