如何在 plot 中制作图例？

Question

我无法找到一种方法来获取我的 plot 的图例。我的数据是一个由 3 列 x、y 和 z 组成的数据框。 x 和 y 表示点的坐标，z 是该点所属的标签（0,1,2,3）。 样本数据：

我需要 plot 散点图，其中包含代表相应 label 的颜色的图例。我已经绘制了散点图，但无法理解如何将图例放入其中。 到目前为止我使用的代码是（dft 是数据框）：

import matplotlib.pyplot as plt
%matplotlib inline
fig = plt.figure(figsize=(8,8))
ax = fig.add_subplot()
ax.scatter(dft['x'] , dft['y'], c=dft['z'], cmap = 'hsv')
plt.show()

我得到的图是：

我需要每种颜色的图例。

Answer 1

您可以尝试使用此代码：

import matplotlib.pyplot as plt
%matplotlib inline
plt.figure(figsize=(8,8))
plt.scatter(dft['x'] , dft['y'], c=dft['z'], cmap = 'hsv')
plt.colorbar()