从Pandas value_counts（）计算提取的值

Question

看起来很简单，但我还没有在其他答案中找到它。

如何为某些变量分配pandas中某些条件的计数，如下例所示：

命令（计算pandas数据帧中列的类别）：

print(data.isBald.value_counts())

输出：

 Bald 2345
 notBald 5432
Name: isBald, dtype: int64

问题是：

isBaldCount = ?  #2345

结果是打印但如何将其分配给变量？

Answer 1

如果您只想要Bald人（!!），请使用 -

isBaldCount = data.isBald.eq('Bald').sum()

另一方面，如果要计算每个值并将其分配给单独的变量，则使用value_counts并分配结果 -

c = data.isBald.value_counts()

isBaldCount = c.loc['Bald'].item()
notBaldCount = c.loc['Not Bald'].item()

Answer 2

outdf = pd.DataFrame(data.isBald.value_counts())
outdf.loc[outdf.index == 'Bald',:]