从矩阵对角线中去除元素

import numpy as np

def remove_diagonal(matrix):
    # 创建一个与原矩阵形状相同的零矩阵
    result = np.zeros_like(matrix)
    for i in range(matrix.shape[0]):
        for j in range(matrix.shape[1]):
            if i != j:
                result[i, j] = matrix[i, j]
    return result

# 示例矩阵
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# 去除对角线元素
result_matrix = remove_diagonal(matrix)
print(result_matrix)

[[0 2 3]
 [4 0 6]
 [7 8 0]]

def remove_diagonal_inplace(matrix):
    for i in range(matrix.shape[0]):
        matrix[i, i] = 0

# 示例矩阵
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# 原地去除对角线元素
remove_diagonal_inplace(matrix)
print(matrix)

[[0 2 3]
 [4 0 6]
 [7 8 0]]