Что не так с этим пониманием вложенного списка? - PullRequest
Новая
       2

Что не так с этим пониманием вложенного списка?

2 голосов
/ 28 ноября 2011 
>>> c = 'A/B,C/D,E/F'
>>> [a for b in c.split(',') for (a,_) in b.split('/')]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <listcomp>
ValueError: need more than 1 value to unpack

Ожидаемый результат: ['A', 'C', 'E'].

Вот как я ожидал бы это сделать, но, видимо, в Python это спереди: 

>>> [a for (a, _) in b.split('/') for b in c.split(',')]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'b' is not defined

Ответы [ 2 ]

5 голосов
/ 28 ноября 2011

Причина вашей неудачи в том, что b.split('/') не дает 2-кортеж. Понимание двойного списка подразумевает, что вы хотите рассматривать декартово произведение как плоский поток, а не матрицу. То есть: 

>>> [x+'/'+y for y in 'ab' for x in '012']
['0/a', '1/a', '2/a', '0/b', '1/b', '2/b']
    # desire output 0,1,2
    # not output 0,1,2,0,1,2

Вы не ищете 6 ответов, вы ищете 3. То, что вы хотите: 

>>> [frac.split('/')[0] for frac in c.split(',')]
['A', 'C', 'E']

Даже если вы используете понимание вложенного списка, вы получите декартово произведение (3x2 = 6) и поймете, что у вас есть дублирующая информация (вам не нужен x2): 

>>> [[x+'/'+y for y in 'ab'] for x in '012']
[['0/a', '0/b'], ['1/a', '1/b'], ['2/a', '2/b']]
    # desire output 0,1,2
    # not [0,0],[1,1],[2,2]

Ниже приведены эквивалентные способы ведения дел. Я вроде как приукрашиваю главное различие между генераторами и списками в этом сравнении.

Декартово произведение в форме списка: 

((a,b,c) for a in A for b in B for c in C)
            #SAME AS#
((a,b,c) for (a,b,c) in itertools.product(A,B,C))
            #SAME AS#
for a in A:
    for b in B:
        for c in C:
            yield (a,b,c)

Декартово произведение в матричной форме: 

[[[(a,b,c) for a in A] for b in B] for c in C]
            #SAME AS#
def fC(c):
    def fB(b,c):
        def fA(a,b,c):
            return (a,b,c)   
        yield [f(a,b,c) for a in A]
    yield [fB(b,c) for b in B]
[fC(c) for c in C]
            #SAME AS#
Cs = []
for c in C:
    Bs = []
    for b in B:
        As = []
        for a in A:
            As += [a]
        Bs += [As]
    Cs += [Bs]
return Cs

Повторное применение функции к списку 

({'z':z} for x in ({'y':y} for y in ({'x':x} for x in 'abc')))
              #SAME AS#
for x in 'abc':
    x2 = {'x':x}
    y2 = {'y':x2}
    z2 = {'z':y2}
    yield z2
              #SAME AS#
def f(x):
    return {'z':{'y':{'x':x}}}
return [f(x) for x in 'abc']     # or map(f,'abc')
4 голосов
/ 28 ноября 2011 
[x.split('/')[0] for x in [a for a in c.split(',')]]

Получает требуемые ['A', 'C', 'E'] из c = 'A / B, C / D, E / F'

...