python-3.x - 如何使用 pandas 数据框中的列来标记气泡图/散点图？

Question

我正在尝试使用 pandas 数据框中的列中的条目来标记从 matplotlib 创建的散点图/气泡图。我见过很多相关的例子和问题(例如 here 和 here )。因此我尝试对情节进行相应的注释。这就是我所做的:

import matplotlib.pyplot as plt
import pandas as pd 
#example data frame
x = [5, 10, 20, 30, 5, 10, 20, 30, 5, 10, 20, 30]
y = [100, 100, 200, 200, 300, 300, 400, 400, 500, 500, 600, 600]
s = [5, 10, 20, 30, 5, 10, 20, 30, 5, 10, 20, 30]
users =['mark', 'mark', 'mark', 'rachel', 'rachel', 'rachel', 'jeff', 'jeff', 'jeff', 'lauren', 'lauren', 'lauren']

df = pd.DataFrame(dict(x=x, y=y, users=users)

#my attempt to plot things
plt.scatter(x_axis, y_axis, s=area, alpha=0.5)
    plt.xlabel(xlabel)
    plt.ylabel(ylabel)
    plt.annotate(df.users, xy=(x,y))
    plt.show()

我使用 pandas datframe 并且不知何故得到了 KeyError - 所以我猜需要一个 dict() 对象？还有其他方法可以使用 pandas 数据框中的条目来标记数据吗？

Answer 1

您可以使用DataFrame.plot.scatter然后通过 DataFrame.iat 循环选择:

ax = df.plot.scatter(x='x', y='y', alpha=0.5)
for i, txt in enumerate(df.users):
    ax.annotate(txt, (df.x.iat[i],df.y.iat[i]))
plt.show()