def transform(X):
m, n = len(X), len(X[0])
return [[X[i][j] for i in range(m)] for j in range(n)]
注意:内部的中括号一定要有,没有内部中括号的二层循环等价于:
for i in range(m):
for j in range(n):
而内部的中括号,调节了两层 for 循环的顺序,先执行外层,再执行内层:
for j in range(n):
for i in range(m):