python读取csv文件并绘图 保存
时间: 2023-11-20 19:53:45 浏览: 177
csv_Foundations_CVSCSV数据文件_python_analytics_
你的代码中有一些问题,可能导致你的图形无法正确绘制。首先,你需要将文件路径中的反斜杠转义,即将"D:\test"改为"D:\\test"。其次,你的筛选条件可能有问题,因为你使用了">"而不是">=",应该将">="改为">="。最后,你需要在绘图之后添加保存图像的代码,例如使用plt.savefig()函数。下面是修改后的代码:
```
import pandas as pd
import glob
import os
import matplotlib.pyplot as plt
file_path = 'D:\\test'
files = glob.glob(os.path.join(file_path, "*.csv"))
for file in files:
df = pd.concat(map(pd.read_csv, file))
df = df[df['battery_percent'] >= 50]
df = df[df['path_select_state'] == "ServerSendPath"]
df = df[['robot_id', 'linear_velocity', 'linear_acceleration', 'battery_percent', 'path_select_state', 'left_wheel_current', 'right_wheel_current']]
df.plot(x="linear_velocity", y="left_wheel_current")
plt.title("Speed reducer", fontsize=16)
plt.xlabel("linear_velocity", fontsize=16)
plt.ylabel("left_wheel_current", fontsize=16)
plt.savefig(file[:-4] + ".png")
plt.show()
```
这段代码将会读取文件夹"D:\\test"下的所有csv文件,筛选出电池电量大于等于50且路径选择状态为"Server***文件中。
阅读全文