关于python的画图问题

  plt.scatter(X, y, c = rideshare_data["abs_hour"])

有没有什么办法,在使用dataframe作为color的时候给他们分配匹配dataframe的label,例如 rideshare_data["abs_hour"]的范围是1-10,在图上给显示1-10分别对应的颜色。

可试试通过构造一个colormap颜色数组来做:

category=data[1].astype(int)
colormap = np.array(['','r', 'y', 'm', 'c','pink','black','blue','gray','purple','green'])

plt.scatter(data[0], data[1], s=100, c=colormap[category])