写出python代码。可以实现通过命令把CSV文件的指定列名生成曲线图,生成的CSV文件自动命名并保存到当前文件夹。通过命令可以查看指定的若干条标题整列的曲线图。 1.CSV文件有100个标题,3000行。 2.可以根据标题生成指定的几条曲线图,一个标题的曲线为一个画布。
时间: 2023-07-15 15:11:15 浏览: 155
以下是实现上述功能的Python代码,需要使用matplotlib和pandas库:
```python
import pandas as pd
import matplotlib.pyplot as plt
import argparse
# 命令行参数解析
parser = argparse.ArgumentParser(description='Generate line chart for specific columns in a CSV file.')
parser.add_argument('filename', type=str, help='CSV file name')
parser.add_argument('-c', '--columns', nargs='+', type=str, help='Column names to generate line chart, separated by space')
args = parser.parse_args()
# 读取CSV文件
df = pd.read_csv(args.filename)
# 循环生成每个指定列的曲线图
for col_name in args.columns:
# 创建画布
fig, ax = plt.subplots()
# 绘制曲线
ax.plot(df[col_name])
# 设置标题和坐标轴标签
ax.set_title(col_name)
ax.set_xlabel('Index')
ax.set_ylabel(col_name)
# 自动保存图片到当前文件夹
fig.savefig(f"{col_name}_line_chart.png")
plt.close(fig)
```
可以通过类似以下的命令使用该脚本:
```bash
python plot_csv.py data.csv -c column1 column2
```
其中`data.csv`为要处理的CSV文件名,`column1`和`column2`为要生成曲线图的列名。该命令会在当前文件夹生成两张图片分别对应`column1`和`column2`的曲线图。
阅读全文