写出python代码。可以实现通过命令把CSV文件的指定列名生成曲线图,生成的CSV文件自动命名并保存到当前文件夹。通过命令可以查看指定的若干条标题整列的曲线图。 1.CSV文件有100个标题,3000行。第一列为行索引,第一行为列索引。 2.可以根据标题生成指定的几条曲线图,一个标题的曲线为一个画布。
时间: 2023-07-15 10:11:22 浏览: 83
以下是实现上述功能的 Python 3 代码,需要安装 pandas 和 matplotlib 库:
```python
import pandas as pd
import matplotlib.pyplot as plt
import os
def plot_csv_column(csv_file, column_names):
# 读取 CSV 文件
df = pd.read_csv(csv_file)
# 遍历指定列名,绘制曲线图
for column_name in column_names:
if column_name not in df.columns:
print(f"{column_name} not found in {csv_file}")
continue
fig, ax = plt.subplots()
ax.plot(df[column_name])
ax.set_xlabel("Index")
ax.set_ylabel(column_name)
plt.show()
# 保存图片
filename = os.path.splitext(csv_file)[0] + f"_{column_name}.png"
fig.savefig(filename)
if __name__ == "__main__":
# 示例:绘制 "Column1" 和 "Column2" 两列的曲线图
plot_csv_column("example.csv", ["Column1", "Column2"])
```
可以在终端中使用以下命令运行代码:
```
python plot_csv_column.py example.csv Column1 Column2
```
其中 `plot_csv_column.py` 为保存上述代码的 Python 文件名,`example.csv` 为要绘制曲线图的 CSV 文件名,`Column1` 和 `Column2` 分别为要绘制的两个列名。可以根据需要修改这些参数。
阅读全文