熊貓：填充矩陣對角線

Question

假設我有一個NxN矩陣（df）。我想用0來填充df的所有對角線。有沒有一個簡單的方法來做到這一點？

Answer 1

你可以使用np.eye(n)：

pd.DataFrame(np.eye(5) * np.array([1, 2, 3, 4, 5]))

Answer 2

考慮數據框df

df = pd.DataFrame(np.random.randint(10, size=[5, 5]), list('ABCDE'), list('VWXYZ')) 
print(df) 

    V W X Y Z 
A 4 6 7 5 4 
B 9 7 3 3 0 
C 1 9 7 8 2 
D 1 9 7 9 9 
E 6 6 6 6 1 

np.eye

df.mul(np.eye(len(df), dtype=int)) 

    V W X Y Z 
A 4 0 0 0 0 
B 0 7 0 0 0 
C 0 0 7 0 0 
D 0 0 0 9 0 
E 0 0 0 0 1 

重構
填充在對角線

z = np.zeros_like(df) 
r = np.arange(len(df)) 
z[r, r] = df.values[r, r] 

pd.DataFrame(z, df.index, df.columns) 

    V W X Y Z 
A 4 0 0 0 0 
B 0 7 0 0 0 
C 0 0 7 0 0 
D 0 0 0 9 0 
E 0 0 0 0 1 

np.triu_indices
填寫關閉對角線

i, j = np.triu_indices(len(df), 1) 

df.values[i, j] = 0 
df.values[j, i] = 0 

df 

    V W X Y Z 
A 4 0 0 0 0 
B 0 7 0 0 0 
C 0 0 7 0 0 
D 0 0 0 9 0 
E 0 0 0 0 1 

Answer 3

一種方法是使用fill_diagonal從舊矩陣填充對角線的新零矩陣。

import numpy as np 

A = np.array([[1,2], [3,4]]) 
A_d = np.zeros_like(A) 
np.fill_diagonal(A_d, A.diagonal()) 

# A_d 
# array([[1, 0], 
#  [0, 4]]) 

Answer 4

np.diag應該是足夠了：

df = pd.DataFrame(np.random.randint(7, size=[6, 6])) 
np.diag(np.diag(df)) 

array([[5, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 0, 0, 0], 
     [0, 0, 0, 1, 0, 0], 
     [0, 0, 0, 0, 3, 0], 
     [0, 0, 0, 0, 0, 3]])