Python在groupby中使用三元运算符

Question

有没有办法在 groupby 条件中使用三元运算符？显然这个语法是无效的。

d = {'name':['bil','bil','bil','jim'],
     'col2': ['acct','law', 'acct2','law'],
     'col3': [1,2,3,55],
     'col4': [1,1,1,2]

    }
df2 = pd.DataFrame(data=d)

df2[['col4']] = df2[['col4']].apply(pd.to_numeric)
df2.groupby(['name','col2'])['col4']\
    .max() if (.max()>30) else ''

Answer 1

您可以使用三元，但我不会向您展示。相反，这里有一个更好的选择 - 屏蔽结果:

v = df2.groupby(['name','col2'])['col4'].max()
v.where(v.gt(30), '')

在 groupby 中使用 lambda 会大大降低其运行速度。