在pandas中绘制分组的DataFrame列

Question

我有一个包含少量列的分组DataFrame，并希望在col2中计算非空值并绘制col1中每组的计数

所以我按col1和col2对数据帧进行了分组 col1组在col2中可以有很多观察值

基本上我想像上面所描述的那样计算每个组的情节关系。

如何在熊猫中做到这一点？

Answer 1

要计算分组非NaN，请使用count：

df    
   col1  col2
0     0   1.0
1     0   NaN
2     1   3.0
3     2   4.0
4     2   NaN
5     2   4.0


df2 = df.groupby('col1').col2.count().to_frame()
df2

      col2
col1      
0        1
1        1
2        2

要绘制地图，只需在.plot上拨打df2。

import matplotlib.pyplot as plt

df2.plot(legend=True)    
plt.show()