我有一個資料框 df，A 列帶有亂數，B 列帶有類別。現在，我使用以下代碼獲得另一列 C：
df.loc[df['A'] >= 50, 'C'] = 1
df.loc[df['A'] < 50, 'C'] = 0
我想獲得一個列“D”，如果遇到 1，它會創建一個序列，否則回傳值 0。所需的資料幀如下所示。
必需的 df
A B C D
17 a 0 0
88 a 1 1
99 a 1 2
76 a 1 3
73 a 1 4
23 b 0 0
36 b 0 0
47 b 0 0
74 b 1 1
80 c 1 1
77 c 1 2
97 d 1 1
30 d 0 0
80 d 1 2
uj5u.com熱心網友回復：
使用GroupBy.cumcount有Series.mask：
df['D'] = df.groupby(['B', 'C']).cumcount().add(1).mask(df['C'].eq(0), 0)
print (df)
A B C D
17 a 0 0
88 a 1 1
99 a 1 2
76 a 1 3
73 a 1 4
23 b 0 0
36 b 0 0
47 b 0 0
74 b 1 1
80 c 1 1
77 c 1 2
97 d 1 1
30 d 0 0
80 d 1 2
或numpy.where：
df['D'] = np.where(df['C'].eq(0), 0, df.groupby(['B', 'C']).cumcount().add(1))
轉載請註明出處，本文鏈接：https://www.uj5u.com/qiye/371448.html