Pandas DataFrame 分組 select 列基於 Z84E2C64F38F78BA3EA5C905AB5A2DA7

Question

我想通過唯一的列名對 dataframe 進行分組，並為每個組計算另一列，但前提是滿足第三列的條件：

這將是三列：

df = pd.DataFrame({'col1':['a','b','b'], 'col2':['c1', 'c2', 'c3'], 'col3':[20,10,20]})

ToDo：如果 col3 == 20，則計算 col2 以按 col1 分組
我設法通過 3 個步驟做到了：

df = df.set_index(['col1'])
df = df.loc[df['col3']==20,'col2']
df.groupby(level=0).count().idxmax()

但我想知道是否可以在一行中使用 lambda 表達式來做到這一點:)

Answer 1

您可以先過濾，然后分組和計數：

df[df['col3']==20].groupby('col1')['col2'].count()

Answer 2

應用版本

df[df['col3']==20].groupby('col1').apply(lambda row: row['col2'].count())