将所有非零矩阵元素设置为1(同时保持其他元素为0)

def set_non_zero_elements_to_one(matrix):
    for i in range(len(matrix)):
        for j in range(len(matrix[0])):
            if matrix[i][j] != 0:
                matrix[i][j] = 1
    return matrix

# 示例矩阵
matrix = [
    [0, 2, 0],
    [3, 0, 1],
    [0, 0, 0]
]

# 调用函数
result = set_non_zero_elements_to_one(matrix)

# 输出结果
print(result)

[
    [0, 1, 0],
    [1, 0, 1],
    [0, 0, 0]
]