Python：Mask 正在用 NaN 替換列值

Question

我試圖用組均值替換 ~15000 列 [列 6:14844] 中的零，同時保留組 label 列 [列 1] 和其他幾個標識列 [列 2:5] 不變。 這是我想出的代碼，它有效，除了它用 NaN 替換了我想跳過 [1:5] 的列

df = df.mask(df.iloc[:, np.r_[1, 6:14844]].eq(0), df.iloc[:, np.r_[1, 6:14844]].groupby('group_label').transform('mean'))

提前致謝。

Answer 1

您只需要分配給列的子集，而不是整個 DataFrame。

df.iloc[:, 6:14844] = (df.iloc[:, 6:14844]
                         .mask(df.iloc[:, 6:14844].eq(0),
                               df.iloc[:, np.r_[1, 6:14844]]
                                 .groupby('group_label')
                                 .transform('mean')))