flyps 发表于 2020-7-21 04:49:23

pandas groupby 计数报错

-import pandas as pd
import numpy as np


a =
b =
df = pd.DataFrame({'a':a, 'b':b})
aa = df.groupby(['b']).size()['A']
bb = df.groupby(['a']).size()['A']

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

BIO-张磊 发表于 2020-7-21 04:49:24

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

雪域谜城 发表于 2020-7-21 05:26:08

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

雪域谜城 发表于 2020-7-21 05:28:31

或者直接加一个判断if 'A' not in a:bb=0
页: [1]
查看完整版本: pandas groupby 计数报错