python - 从数据框中删除匹配的索引值

Question

df:

         0         1         2 
0 0.0481948 0.1054251 0.1153076 
1 0.0407258 0.0890868 0.0974378 
2 0.0172071 0.0376403 0.0411687
etc.

我想删除数据框的 x 和 y 标题/值相等的所有值，因此，我的预期输出将类似于:

         0         1         2 
0 NaN       0.1054251 0.1153076 
1 0.0407258 NaN       0.0974378 
2 0.0172071 0.0376403 NaN
etc.

如图所示，(0,0)、(1,1)、(2,2) 等的值已被删除/替换。

我想到了如下遍历索引:

for (idx, row) in df.iterrows():
    if (row.index) == ???

但不知道从哪里继续，甚至不知道这是否是正确的方法

Answer 1

可以设置对角线:

In [11]: df.iloc[[np.arange(len(df))] * 2] = np.nan

In [12]: df
Out[12]:
          0         1         2
0       NaN  0.105425  0.115308
1  0.040726       NaN  0.097438
2  0.017207  0.037640       NaN