將 pandas DataFrame 有條件地分成兩組

Question

我試圖根據該列最后一行中的值是否大於 0 將數據集分成兩組。

為了顯示：

圖片

這使我可以將 1,000 個客戶分成兩組。

pivot_data_credit = data.pivot_table(values='nr_credit_applications', index='yearmonth', columns=['client_nr'])

pivot_data_credit.loc[:,'Credit Total'] = pivot_data_credit.sum(axis=1)

pivot_data_credit.loc['TotalRow'] = pivot_data_credit.sum()

謝謝你的幫助。

Answer 1

您可以在最后一行使用iloc進行屏蔽，並使用帶有loc方法的屏蔽來過濾列：

positive_mask = pivot_data_credit.iloc[-1, :] > 0
positive_clients = pivot_data_credit.loc[:, positive_mask]
non_positive_clients = pivot_data_credit.loc[:, ~positive_mask]