我有以下數據框：account_num trans_cdi1 c1 d 3 d3 c3 d它是更大數據框的一部分。首先，我想對 account_num 列進行分組，然后刪除組中 trans_cdi 列中以“d”開頭的行。所以在這種情況下，新的數據框看起來像這樣：account_num trans_cdi1 c1 d 3 c3 d我那樣試過，但它刪除了整個組：df = df.groupby("account_numb").filter(lambda x: x["trans_cdi"].iloc[0] != "d")謝謝！
2 回答
躍然一笑
TA貢獻1826條經驗 獲得超6個贊
您可以先獲取每組的行數，然后計算 if not(row count equals 0 and row equals d) 它處理每組中不等于 d 的第一行。
idx = df.groupby('account_num').cumcount()
out = df[~(idx.eq(0) & df['trans_cdi'].eq('d'))]
print(out)
account_num trans_cdi
0 1 c
1 1 d
3 3 c
4 3 d
三國紛爭
TA貢獻1804條經驗 獲得超7個贊
filter由于某種原因我無法使用它，所以我transform改用了索引，它返回了預期的輸出：
df2 = df[df.groupby('account_num')['trans_cdi'].transform(lambda x: (x.iloc[0] != "d") | (x.index != x.index[0]))]
account_num trans_cdi
0 1 c
1 1 d
3 3 c
4 3 d
這會概括您的想法嗎？
添加回答
舉報
0/150
提交
取消