如何从矩阵打印行/列/对角线

# 矩阵打印行/列/对角线示例代码

def print_matrix(matrix):
    rows = len(matrix)
    cols = len(matrix[0])

    # 打印行
    for i in range(rows):
        for j in range(cols):
            print(matrix[i][j], end=' ')
        print()

    print()

    # 打印列
    for j in range(cols):
        for i in range(rows):
            print(matrix[i][j], end=' ')
        print()

    print()

    # 打印主对角线
    for i in range(rows):
        for j in range(cols):
            if i == j:
                print(matrix[i][j], end=' ')
        print()

    print()

    # 打印副对角线
    for i in range(rows):
        for j in range(cols):
            if i + j == rows - 1:
                print(matrix[i][j], end=' ')
        print()

# 测试示例
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
print_matrix(matrix)