Itereren over een tweedimensionale pythonlijst

Ik heb een array met 2 dimensies gemaakt zoals:

rows =3
columns= 2
mylist = [[0 for x in range(columns)] for x in range(rows)]
for i in range(rows):
    for j in range(columns):
        mylist[i][j] = '%s,%s'%(i,j)
print mylist

Het afdrukken van deze lijst geeft een uitvoer:

[  ['0,0', '0,1'], ['1,0', '1,1'], ['2,0', '2,1']   ]

waarbij elk lijstitem een tekenreeks is met de notatie ‘rij,kolom’

Nu ik deze lijst heb gezien, wil ik deze in de volgorde doorlopen:

'0,0'
'1,0'
'2,0'
'0,1'
'1,1'
'2,1'

dat is herhalen door de eerste kolom, dan door de tweede kolom, enzovoort. Hoe doe ik het met een lus?

Deze vraag heeft betrekking op de pure python-lijst, terwijl de vraag die als hetzelfde is gemarkeerd, betrekking heeft op numpy-arrays. Ze zijn duidelijk anders


Antwoord 1, autoriteit 100%

Gebruik zipen itertools.chain. Iets als:

>>> from itertools import chain
>>> l = chain.from_iterable(zip(*l))
<itertools.chain object at 0x104612610>
>>> list(l)
['0,0', '1,0', '2,0', '0,1', '1,1', '2,1']

Antwoord 2, autoriteit 96%

dezelfde manier waarop u de vulling deed, maar keer de indexen om:

>>> for j in range(columns):
...     for i in range(rows):
...        print mylist[i][j],
... 
0,0 1,0 2,0 0,1 1,1 2,1
>>> 

Antwoord 3, autoriteit 94%

Dit is de juiste manier.

>>> x = [ ['0,0', '0,1'], ['1,0', '1,1'], ['2,0', '2,1'] ]
>>> for i in range(len(x)):
        for j in range(len(x[i])):
                print(x[i][j])
0,0
0,1
1,0
1,1
2,0
2,1
>>> 

Antwoord 4, autoriteit 21%

>>> mylist = [["%s,%s"%(i,j) for j in range(columns)] for i in range(rows)]
>>> mylist
[['0,0', '0,1', '0,2'], ['1,0', '1,1', '1,2'], ['2,0', '2,1', '2,2']]
>>> zip(*mylist)
[('0,0', '1,0', '2,0'), ('0,1', '1,1', '2,1'), ('0,2', '1,2', '2,2')]
>>> sum(zip(*mylist),())
('0,0', '1,0', '2,0', '0,1', '1,1', '2,1', '0,2', '1,2', '2,2')

Antwoord 5, autoriteit 12%

zipzal de lijst transponeren, daarna kun je de outputs samenvoegen.

In [3]: zip(*[ ['0,0', '0,1'], ['1,0', '1,1'], ['2,0', '2,1'] ])
Out[3]: [('0,0', '1,0', '2,0'), ('0,1', '1,1', '2,1')]

Antwoord 6, autoriteit 6%

Ref: ingebouwde zip-functie

zip()in combinatie met de operator *kan worden gebruikt om een lijst te unzip

unzip_lst = zip(*mylist)
for i in unzip_lst:
    for j in i:
        print j

Antwoord 7, autoriteit 3%

>>> [el[0] if i < len(mylist) else el[1] for i,el in enumerate(mylist + mylist)]
['0,0', '1,0', '2,0', '0,1', '1,1', '2,1']

Other episodes