帮我续写代码,以STU,THU,BKU为数据创建一个折线图
时间: 2024-03-05 15:53:31 浏览: 130
好的,以下是使用 matplotlib 库绘制上海交通大学、清华大学和北京大学在 ICPC 中的表现的折线图的示例代码:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取数据
data = pd.read_csv('icpc-full.csv')
# 去除多余数据
data = data.drop(['Date','Host','City','Venue','Team','Contestant 1','Contestant 2','Contestant 3','Score','Total'], axis=1)
# 查找数据
STU = data[data['University'] == 'Shanghai Jiao Tong University']
THU = data[data['University'] == 'Tsinghua University']
BKU = data[data['University'] == 'Peking University']
# 绘制折线图
fig, ax = plt.subplots()
ax.plot(STU['Year'], STU['Rank'], color='red', label='Shanghai Jiao Tong University')
ax.plot(THU['Year'], THU['Rank'], color='blue', label='Tsinghua University')
ax.plot(BKU['Year'], BKU['Rank'], color='green', label='Peking University')
ax.set_xlabel('Year')
ax.set_ylabel('Rank')
ax.legend()
plt.show()
```
上述代码使用 `plot()` 方法绘制折线图,其中 `STU['Year']` 表示上海交通大学参加 ICPC 的年份,`STU['Rank']` 表示上海交通大学在 ICPC 中的排名,其他两个学校类似。`color` 参数可以用来设置折线的颜色,`label` 参数则可以用来为图例设置标签。最后使用 `legend()` 方法添加图例,并使用 `show()` 方法显示图形。您可以根据需要进行调整。
阅读全文