Pythonで転置行列

以下の転置行列を求める問題を考える．

zipを使う

zipを使うとすごくきれいに書ける．

H, W = map(int, input().split())
A = [list(map(int, input().split())) for _ in range(H)]

B = list(zip(*A))
for b in B:
  print(*b)

【参考】【解説 実況】ABC237 AからD【かつっぱ】 - YouTube

素直に書く

素直に書いても，まあ簡単．

H, W = map(int, input().split())
A = [list(map(int, input().split())) for _ in range(H)]

B = [[A[j][i] for j in range(H)] for i in range(W)]
for b in B:
  print(*b)

例

• B - Matrix Transposition
• C - RANDOM (ABC279C - RANDOM - 競プロはじめました)