Python写代码,通过传参标题名,查看多个指定列的曲线
时间: 2023-06-17 17:08:49 浏览: 78
显示多条数值曲线的类
5星 · 资源好评率100%
假设你有一个名为 data.csv 的数据文件,数据文件中包含多列数据,你可以使用 pandas 库来读取数据文件,并通过 matplotlib 库来绘制曲线。下面是一个示例代码,可以通过传参标题名和列名列表来查看多个指定列的曲线:
```python
import argparse
import pandas as pd
import matplotlib.pyplot as plt
# 创建命令行参数解析器
parser = argparse.ArgumentParser(description='Plot multiple columns of data.')
# 添加标题名参数
parser.add_argument('title', type=str, help='Title of the plot')
# 添加列名参数
parser.add_argument('columns', nargs='+', type=str, help='Columns to plot')
# 解析命令行参数
args = parser.parse_args()
# 读取数据文件
data = pd.read_csv('data.csv')
# 获取要绘制的列数据
plot_data = data[args.columns]
# 绘制曲线
plot_data.plot()
# 设置标题
plt.title(args.title)
# 显示图形
plt.show()
```
你可以在命令行中输入以下命令来运行代码:
```bash
python plot_data.py 'Plot Title' column1 column2 column3
```
其中,'Plot Title' 是你想要设置的标题名,column1、column2 和 column3 是你要绘制的列名。你可以根据需要修改代码来适应你的数据文件和绘图需求。
阅读全文