改进代码,增加功能可选任意几个标题生成曲线。代码如下:import pandas as pd import matplotlib.pyplot as plt # 读取CSV文件 data = pd.read_csv('your_file_path.csv') # 获取所有标题 titles = data.columns.tolist() # 用户选择要绘制的标题 selected_title = input("Please enter the title you want to plot: ") # 获取相应标题的数据列 selected_data = data[selected_title] # 绘制曲线图 plt.plot(selected_data) plt.title(selected_title) plt.xlabel("Index") plt.ylabel("Value") plt.show()
时间: 2023-07-14 11:14:20 浏览: 269
可以考虑添加一个多选框,让用户选择要绘制的标题,而不是只能选择一个。代码如下:
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.widgets import CheckButtons
# 读取CSV文件
data = pd.read_csv('your_file_path.csv')
# 获取所有标题
titles = data.columns.tolist()
# 多选框
options = [plt.axes([0.025*i, 0.5, 0.1, 0.1], facecolor='lightgoldenrodyellow') for i in range(len(titles))]
check = CheckButtons(options, titles, actives=[False for _ in titles])
# 获取用户选择的标题
selected_titles = []
def func(label):
if label in selected_titles:
selected_titles.remove(label)
else:
selected_titles.append(label)
check.on_clicked(func)
# 绘制曲线图
def plot_selected_data(selected_titles):
for title in selected_titles:
selected_data = data[title]
plt.plot(selected_data, label=title)
plt.xlabel("Index")
plt.ylabel("Value")
plt.legend()
plt.show()
# 调用函数
plot_selected_data(selected_titles)
阅读全文