矩阵转置 -- Python

'''
转置矩阵
'''

# 我写的
def trans_matrix(a):
    m = len(a)
    n = len(a[0])
    b = [[0] * m for i in range(n)]
    for i in range(m):
        for j in range(n):
            b[j][i] = a[i][j]
    return b

# 别人写的
def trans_matrix1(a):
    b = [[] for i in a[0]]
    # 拿到每一行
    for i in a:
        # 每行取前n个，每行append
        for j in range(len(i)):
            b[j].append(i[j])
    return b


if __name__ == '__main__':
    a = [[1, 2], [4, 5], [7, 8]]
    a_t = trans_matrix1(a)
    print(a_t)