python - 用单元格位置替换 bool 数据框中的所有 True

Question

我有一个 bool 数据框，想用单元格的位置(作为元组)替换 True 单元格。例子:

import pandas as pd

df = pd.DataFrame({'A': [True, False, False],
                  'B': [False, True, True]})

我尝试了 df.mask(df, df.index) 的修改版本来自 here (例如，尝试 iloc )但它不起作用。

Answer 1

让我们试试stack将数据框 reshape 为系列，然后使用 bool 索引与 loc填充使用 np.argwhere 获得的索引, 最后使用 unstack reshape 回数据框:

m = df.stack()
m.loc[m] = pd.Series(map(tuple, np.argwhere(df.to_numpy())), index=m[m].index)
out = m.unstack()

或者你也可以试试 np.argwhere获取索引并使用 iat 进行索引设置 True 中的值具有相应索引的单元格:

out = df.astype(object)
for r, c in np.argwhere(df.to_numpy()):
    out.iat[r, c] = (r, c)

结果:

print(out)

        A       B
0  (0, 0)   False
1   False  (1, 1)
2   False  (2, 1)