如何识别矩阵中缺少的行/列，并在其中放置一个零？(对称不匹配)

import numpy as np

def insert_zero(matrix):
    # 获取矩阵的行数和列数
    rows, cols = matrix.shape

    # 初始化一个包含零元素的行/列
    zero_row = np.zeros(cols)
    zero_col = np.zeros(rows)[:, np.newaxis]

    # 遍历矩阵的行和列，找出缺失的行/列，并插入零元素
    for i in range(rows):
        if np.count_nonzero(matrix[i, :]) == 0:
            matrix = np.insert(matrix, i, zero_row, axis=0)
    for j in range(cols):
        if np.count_nonzero(matrix[:, j]) == 0:
            matrix = np.insert(matrix, j, zero_col, axis=1)

    return matrix

# 示例输入矩阵
input_matrix = np.array([[1, 2, 3],
                         [4, 0, 6],
                         [7, 8, 9]])

# 插入零元素后的矩阵
output_matrix = insert_zero(input_matrix)
print(output_matrix)