试题 算法提高 矩阵转置

思路：创建一个新的矩阵，利用B[j][i] = A[i][j]得到新的矩阵的值，输出即可，详看代码。
测试样例有个错误， 只有90分。

资源限制
时间限制：1.0s   内存限制：256.0MB
【问题描述】

　　给定一个n×m矩阵相乘，求它的转置。其中1≤n≤20，1≤m≤20，矩阵中的每个元素都在整数类型（4字节）的表示范围内。
【输入格式】
　　第一行两个整数n和m；
　　第二行起，每行m个整数，共n行，表示n×m的矩阵。数据之间都用一个空格分隔。
【输出格式】

　　共m行，每行n个整数，数据间用一个空格分隔，表示转置后的矩阵。
样例输入
2 4
34 76 -54 7
-4 5 23 9
样例输出
34 -4
76 5
-54 23
7 9

代码：

def fun(s,t,n,m):
    for i in range(n):
        for j in range(m):
            T[j][i] = s[i][j]
    return t 

if __name__ == "__main__":
    nums = list(map(int,input().split()))
    n = nums[0]
    m = nums[1]
    if 1<=n<=20 and 1<=m<=20:
        s = []
        T = [[0]* n for i in range(m)] # m*n的矩阵且值都为0
        i = 0 
        while i < n:
            s.append(list(map(int,input().split())))
            i += 1
        T = fun(s,T,n,m)
        # 打印
        for i in range(len(T)):
                count = 0 # 计数
                for j in range(len(T[0])):
                    print(T[i][j],end=' ')
                    count += 1
                    if count == len(T[0]): # 判断一行是否打印完
                        print()