写出python代码。可以实现通过命令把CSV文件的所有列名生成曲线图里,在一张画布里能够显示所有列的曲线图。生成的CSV文件自动命名并保存到当前文件夹。 1.CSV文件有80个标题,3000行。第一列为横轴坐标。把80条线一起显示出来。 2.可以根据标题生成所有列的曲线图。每一列曲线图的颜色都不同。
时间: 2023-07-15 11:13:39 浏览: 86
python导入csv文件绘制曲线图
以下是实现功能的Python代码:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取CSV文件
file_name = input("请输入CSV文件名:") + '.csv'
df = pd.read_csv(file_name)
# 获取所有列名
columns = list(df.columns)
# 删除第一列(横轴坐标)
del columns[0]
# 生成曲线图
fig, ax = plt.subplots()
for i in columns:
ax.plot(df.iloc[:, 0], df[i], label=i)
# 图例显示
ax.legend()
plt.show()
# 自动保存CSV文件
output_file_name = file_name.split('.')[0] + '_output.csv'
df.to_csv(output_file_name, index=False)
print("CSV文件已保存为:", output_file_name)
```
这个代码可以实现以下功能:
1. 通过命令输入CSV文件名,读取CSV文件。
2. 获取CSV文件的所有列名,并删除第一列。
3. 根据标题生成所有列的曲线图,并设置每一列曲线图的颜色不同。
4. 显示所有曲线图,并生成CSV文件的输出文件。
5. 输出文件名自动将输入文件名替换为“_output.csv”。
阅读全文