写出python代码。可以实现通过命令把CSV文件的指定列名生成曲线图,生成的CSV文件自动命名并保存到当前文件夹。通过命令可以查看指定的若干条标题整列的曲线图。 1.CSV文件有100个标题,3000行。第一列为行索引,第一行为列索引。 2.可以根据标题生成指定的几条曲线图,一个标题的曲线为一个画布。
时间: 2023-06-27 22:05:58 浏览: 37
以下是实现上述功能的Python代码:
```python
import matplotlib.pyplot as plt
import pandas as pd
import argparse
import os
# 命令行参数解析
parser = argparse.ArgumentParser(description='Plot curves from CSV file.')
parser.add_argument('filename', type=str, help='CSV filename')
parser.add_argument('columns', nargs='+', type=str, help='Column names to plot')
args = parser.parse_args()
# 读取CSV文件
df = pd.read_csv(args.filename)
# 绘制曲线图
for col in args.columns:
if col in df.columns:
plt.plot(df[col], label=col)
plt.legend()
# 自动命名并保存文件
filename = os.path.splitext(args.filename)[0] + '_{}.png'.format('-'.join(args.columns))
plt.savefig(filename)
# 显示曲线图
plt.show()
```
可以通过以下命令使用该脚本生成曲线图:
```
python plot_csv.py data.csv col1 col2 col3
```
其中,`data.csv`为要绘制的CSV文件,`col1 col2 col3`为要绘制的列名,可以指定多个列名。该命令将会生成一个名为`data_col1-col2-col3.png`的图像文件,并在屏幕上显示出来。