将 dataframe 中的上下三角形值替换为零，或仅保留对角线值

Question

我有以下 DataFrame 作为玩具示例：

a = [5,2,6,8]
b = [2,10,19,16]
c = [3,8,15,17]
d = [3,8,12,20]
df  = pd.DataFrame([a,b,c,d], columns = ['a','b','c','d'])
df

我想创建一个新的 DataFrame df1 ，它只保留对角线元素并将上下三角形值转换为零。

我的最终数据集应如下所示：

    a   b   c   d
0   5   0   0   0
1   0   10  0   0
2   0   0   15  0
3   0   0   0   20

Answer 1

您可以使用numpy.diag ：

df = pd.DataFrame(data=np.diag(np.diag(df)), columns=df.columns)
print(df)

Output

   a   b   c   d
0  5   0   0   0
1  0  10   0   0
2  0   0  15   0
3  0   0   0  20

Answer 2

import pandas as pd

def diag(df):
    res_df = pd.DataFrame(0, index=df.index, columns=df.columns)
    for i in range(min(df.shape)): res_df.iloc[i, i] = df.iloc[i, i]
    return res_df