python - 计算从 Pandas value_counts() 中提取的值

Question

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

如何将 pandas 中某些条件的计数分配给某个变量，如下例所示:

命令(计算 pandas dataframe 中列的类别):

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()