我有一个数据框,我想做一个散点图。
数据帧看起来像:
year length Animation
0 1971 121 1
1 1939 71 1
2 1941 7 0
3 1996 70 1
4 1975 71 0
我希望散点图中的点是不同的颜色,这取决于动画行中的值。
所以动画=1=黄色
动画=0=黑色
或者类似的东西
我试着做了以下事情:
dfScat = df[['year','length', 'Animation']]
dfScat = dfScat.loc[dfScat.length < 200]
axScat = dfScat.plot(kind='scatter', x=0, y=1, alpha=1/15, c=2)
这就产生了一个滑块,很难区分两者的区别。
最佳答案
在c
中使用scatter
参数
df.plot.scatter('year', 'length', c='Animation', colormap='jet')