import sys

while True:
  a=[]
  res=" "
  a=sys.stdin.read().split()
  if len(a):
    row=int(a[0])
    col=int(a[1])
    a.pop(0)
    for i in range(0,col):
     for j in range(i,(i+col*(row-1))+1,col):
      res+=a[j]+" "
      res=res[:1]
      res+="\n"
      print(res)
   else:
     break

每組測資可以用

for line in sys.stdin:
    [row, col] = [int(x) for x in line.strip().split()]
    # TODO

來開頭，會單純一點。

把 a 做成 nested list，能直接取 a[r][c]，會比較方便。