Как перенести массив 2D-списка с использованием циклов в python?

Question

скажите a = [[1, 1, 1, 6], [0, 2, -1, 3], [4, 0, 10, 42]] и я хочу перенести его на a = [[1, 0,4], [1,2,0], [1, -1,10], [6,3,42]] с использованием петель в python. Текущий код, который у меня есть:

def transpose(a): 
    s = [] 
    for row in range(len(a)): 
     for col in range(len(a)): 
      s = s + [a[col][row]] 
return s 

Но это дает мне выход:

[1, 0, 4, 1, 2, 0, 1, -1, 10] 

Вместо этого:

[[1,0,4], [1,2,0], [1,-1,10], [6,3,42]] 

Может кто-нибудь мне помочь? Я все еще новичок в этом и не понимаю, почему это не работает. Спасибо!

Answer 1

Использование zip()

>>> a = [[1, 1, 1, 6], [0, 2, -1, 3], [4, 0, 10, 42]] 
>>> [list(x) for x in zip(*a)] 
[[1, 0, 4], [1, 2, 0], [1, -1, 10], [6, 3, 42]] 

zip(*a) распаковывает три суб-листы в a и объединяет их поэлементно. Смысл, первые элементы каждого из трех подписок объединены вместе, а другие элементы объединены вместе и так далее. Но zip() возвращает кортежи вместо списков, как вы хотите в своем выходе. Как это:

>>> zip(*a) 
[(1, 0, 4), (1, 2, 0), (1, -1, 10), (6, 3, 42)] 

[list(x) for x in zip(*a)] преобразует каждый из кортежей списков, давая выходной сигнал так, как вам это нужно.

Answer 2

Вот решение, которое основано на коде:

def transpose(a): 
    s = [] 
    # We need to assume that each inner list has the same size for this to work 
    size = len(a[0]) 
    for col in range(size): 
     inner = [] 
     for row in range(len(a)): 
      inner.append(a[row][col]) 
     s.append(inner) 
    return s 

Если определить внутренний список для внутреннего контура, ваш вывод заключается в следующем:

[[1, 0, 4], [1, 2, 0], [1, -1, 10], [6, 3, 42]] 

Answer 3

Если вы ищете решение без какой-либо фантастической функции. Вы можете достичь его, используя список понимание как:

>>> a = [[1, 1, 1, 6], [0, 2, -1, 3], [4, 0, 10, 42]] 
>>> sublist_size = len(a[0]) 
>>> [[item[i] for item in a] for i in range(sublist_size)] 
[[1, 0, 4], [1, 2, 0], [1, -1, 10], [6, 3, 42]] 

Однако, самый простой способ заключается в использовании zip():

>>> list(zip(*a)) # for Python 3, OR, just zip(*a) in Python 2 
[(1, 0, 4), (1, 2, 0), (1, -1, 10), (6, 3, 42)]