实现此需求的主要方法是使用 Pandas 库来对数据进行预处理和分组,并使用 Matplotlib 库来绘制线图。以下是一个简单的示例:
假设我们有一个数据框 df,其中包含三列 x、y、color,表示两个连续的坐标点和一组唯一的颜色,我们希望按照颜色分组并绘制多个线:
import pandas as pd
import matplotlib.pyplot as plt
# 创建示例数据框
data = {
'x': [1, 2, 3, 4, 5, 6, 7, 8],
'y': [5, 2, 8, 2, 3, 9, 1, 5],
'color': ['red', 'blue', 'green', 'red', 'blue', 'green', 'red', 'green']
}
df = pd.DataFrame(data)
# 按照颜色分组,绘制多条线
fig, ax = plt.subplots()
for color, group in df.groupby('color'):
group.plot(x='x', y='y', ax=ax, label=color)
plt.show()
上述代码首先创建一个数据框,并按照颜色分组。然后使用 for 循环遍历每个颜色组,并在图表上绘制对应的线。最后,使用 plt.show() 函数显示图表。
上一篇:按照唯一值分组并计算每个组的总和
下一篇:按照唯一值和频率显示的R代码