在Python中,如何在CSV数据绘图中添加一个没有标题的图例?

2024-06-02 08:47:38 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个csv文件,其中包含要打印的值。该文件没有如下所示的标题

          0.95744324  0.09625244  7.9512634
0       0.840118    0.153717   7.841126
1       0.646194    0.292572   7.754929
2       0.492966    0.452988   7.829147
3       0.291855    0.646912   7.991959
4       0.279877    0.716354   8.039841
...          ...         ...        ...

我可以用下面的代码将每一列绘制成一个单独的线,但我想为相应的列/线添加一个x、y、z尺寸的图例。我不知道该怎么做,因为我现在用的是图例“x”中的所有键。我不能修改csv文件,所以我应该在代码中添加标题,然后分别绘制每一列吗

aPlot = pd.read_csv('accl/a.csv')


plt.figure()
plt.plot(aPlot, label = "x")
plt.xlabel("time")
plt.ylabel("acceleration[m/s^2")
plt.legend(loc="upper left")
plt.show

Tags: 文件csv代码标题readplot尺寸绘制
1条回答
网友
1楼 · 发布于 2024-06-02 08:47:38

由于CSV文件没有标题,您可以通过传递names参数来指定列名

然后可以使用数据框进行绘图,图例将正确无误:

import matplotlib.pyplot as plt
import pandas as pd
 
aPlot = pd.read_csv('input.csv', names=['col1', 'col2', 'col3'])

aPlot.plot()
plt.xlabel("time")
plt.ylabel("acceleration[m/s^2")
plt.legend(loc="upper left")
plt.show()        

给你:

matplotlib plot output

相关问题 更多 >