在 Dataframe 中设置对角线值

Question

已经问过类似的问题，但我仍然无法提出解决方案。

我有一个这样形状的熊猫数据框：

 1     0     0     0     0
 1     0     0     0     0
 1     0     0     0     0
 1     0     0     0     0
-1     0     0     0     0
-1     0     0     0     0
-1     0     0     0     0
-1     0     0     0     0

我想把它做成这样的：

 1     1     0     0     0
 1     0     1     0     0
 1     0     0     1     0
 1     0     0     0     1
-1     1     0     0     0
-1     0     1     0     0
-1     0     0     1     0
-1     0     0     0     1

我试过 np.fill_diagonal(df.values,1) 但没有用。

Answer 1

假设第一列确实是一列并且您想忽略它：

a = df.to_numpy()
idx = np.arange(a.shape[0])
col = np.arange(a.shape[0])%(a.shape[1]-n_ignore)+n_ignore
a[idx, col] = 1

df2 = pd.DataFrame(a, index=df.index, columns=df.columns)