折线图中的格式标记形状 matplotlib

Question

我是 python 的新手，希望自定义折线图。

我正在使用 power bi 中的 python 视觉对象来绘制网球比赛中的分数。 我正在使用以下有效的代码来生成一个简单的图表：

import matplotlib.pyplot as plt
dataset.plot(kind='line', x = 'Index', y = 'Running total', marker = 'o')
plt.show()

在我的数据集中，我还有另一列 [Shot]，显示用于完成得分的击球类型（例如 ace、error、winner 等）。 我想根据镜头类型更改标记的形状。 例如圆圈代表 ace，x 代表错误，三角形代表赢家）。

我怀疑我需要使用 for 循环，但我正在努力弄清楚如何构造它。

任何帮助或指导将不胜感激。

TIA！

尝试了包括 for 循环的不同方法，但努力使语法正确

指数	运行总计	乐谱类型
1个	0	强制错误
2个	1个	优胜者
3个	2个	高手
4个	1个	双重故障
5个	2个	强制错误
6个	3个	高手
7	4个	强制错误
8个	3个	强制错误
9	2个	优胜者
10	1个	双重故障

Answer 1

感谢您的更新和数据。 使用您提供的数据，您可以通过散点图 plot 实现此目的。

这是一个例子：

import pandas as pd
import matplotlib.pyplot as plt

# create the dataframe
data = {'Index': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
        'Running Total': [0, 1, 2, 1, 2, 3, 4, 3, 2, 1],
        'Score type': ['Forced error', 'Winner', 'Ace', 'Double fault', 
                       'Forced error', 'Ace', 'Forced error', 'Forced error', 
                       'Winner', 'Double fault']}

df = pd.DataFrame(data)

# plot the line chart
plt.plot(df['Index'], df['Running Total'], '-')

# plot the scatter chart
categories = df['Score type'].astype("category").cat.categories
for i, category in enumerate(categories):
    plt.scatter(df[df['Score type'] == category]['Index'],
                df[df['Score type'] == category]['Running Total'],
                label=category, c=f"C{i}")

plt.xlabel('Index')
plt.ylabel('Running Total')
plt.legend()
plt.show()

这是结果：

你可以看到我把图表分成两部分。

1. 折线图
2. 分散 plot