如何从数据框中的列中获取点的标签以用于 python 中的散点图 - TypeScript

在 Python 中，在绘制散点图时，我们通常需要给每个数据点添加标签以便于读取。而获取这些标签通常需要从数据框中读取，这篇文章将介绍如何从数据框中获取点的标签以用于 Python 中的散点图。

1. 数据准备

假设我们有以下数据：

import pandas as pd

data = {
    'x': [1, 2, 3, 4, 5],
    'y': [2, 4, 6, 8, 10],
    'label': ['A', 'B', 'C', 'D', 'E']
}

df = pd.DataFrame(data)

其中，'x' 和 'y' 为散点图的坐标，'label' 为每个点的标签。

2. 绘制散点图

使用 Python 中的 matplotlib 库可以绘制散点图。

import matplotlib.pyplot as plt

plt.scatter(df['x'], df['y'])
plt.show()

运行以上代码将得到一个没有标签的散点图。

3. 获取标签并添加到散点图上

我们可以使用 for 循环遍历数据框中的每一行来获取标签，并将其添加到散点图上。

for i in range(len(df)):
    plt.scatter(df.loc[i]['x'], df.loc[i]['y'], label=df.loc[i]['label'])

plt.legend()
plt.show()

以上代码将从数据框中获取每个点的标签，并将它们添加到散点图上。运行后将得到一个带标签的散点图。

4. 完整代码

import pandas as pd
import matplotlib.pyplot as plt

data = {
    'x': [1, 2, 3, 4, 5],
    'y': [2, 4, 6, 8, 10],
    'label': ['A', 'B', 'C', 'D', 'E']
}

df = pd.DataFrame(data)

for i in range(len(df)):
    plt.scatter(df.loc[i]['x'], df.loc[i]['y'], label=df.loc[i]['label'])

plt.legend()
plt.show()

以上代码展示了如何从数据框中获取点的标签以用于 Python 中的散点图。