[已解决]pandas groupby 计数报错

flyps

-

1. import pandas as pd
   
2. import numpy as np
   
3. 
   
4. 
   
5. a = [1,2,3]
   
6. b = [4,5,'A']
   
7. df = pd.DataFrame({'a':a, 'b':b})
   
8. aa = df.groupby(['b']).size()['A']
   
9. bb = df.groupby(['a']).size()['A']

复制代码

运行到最后一行的代码会报错，这是因为a列没有'A',
请问要如何改，当a列没有'A'，不会出错，并bb赋值为0，谢谢

最佳答案

月排行榜 / 总排行榜

BIO-张磊

2020-7-21 04:49:24

1. try:
   
2.      bb = df.groupby(['a']).size()['A']
   
3. except KeyError:
   
4.      bb = 0

复制代码

跳转到最佳答案楼层

BIO-张磊

1. try:
   
2.      bb = df.groupby(['a']).size()['A']
   
3. except KeyError:
   
4.      bb = 0

复制代码

雪域谜城

加一个try,finally语句就可以了吧

雪域谜城

或者直接加一个判断if 'A' not in a:bb=0