У меня есть два генератора g1 и g2
for line in g1:
print line[0]
[a, a, a]
[b, b, b]
[c, c, c]
for line1 in g2:
print line1[0]
[1, 1, 1]
[2, 2, 2]
[3, 3, 3]
for line in itertools.chain(g1, g2):
print line[0]
[a, a, a]
[b, b, b]
[c, c, c]
[1, 1, 1]
[2, 2, 2]
[3, 3, 3]
Как
Я получаю вывод как:
[a, a, a], [1, 1, 1]
[b, b, b], [2, 2, 2]
[c, c, c], [3, 3, 3]
или
[a, a, a, 1, 1, 1]
[b, b, b, 2, 2, 2]
[c, c, c, 3, 3, 3]
Спасибо за вашу помощь.
первый случай: используйте
for x, y in zip(g1, g2):
print(x[0], y[0])
второй случай: используйте
for x, y in zip(g1, g2):
print(x[0] + y[0])
Конечно, вы можете использовать itertools.izip для версии генератора. Вы получаете генератор автоматически, если используете zip в Python 3 и выше.
Вы можете получить пару вещей (ваш первый запрос), используя zip(g1, g2). Вы можете присоединиться к ним (ваш второй запрос), выполнив [a + b for a, b in zip(g1, g2)].
Почти эквивалентно, вы можете использовать map. Используйте map(None, g1, g2) для создания списка пар и map(lambda x, y: x + y, g1, g2) для объединения пар вместе.
В ваших примерах ваши генераторы каждый раз производят список или кортеж, из которых вас интересует только первый элемент. Я бы просто сгенерировал то, что вам нужно, или предварительно обработал их перед тем, как накладывать или отображать их. Например:
g1 = (g[0] for g in g1)
g2 = (g[0] for g in g2)
В качестве альтернативы вы можете применить [0] на карте. Здесь два случая:
map(lambda x, y: (x[0], y[0]), g1, g2)
map(lambda x, y: x[0] + y[0], g1, g2)
Вы можете использовать itertools.izip, например
g1=([s]*3 for s in string.ascii_lowercase)
g2=([s]*3 for s in string.ascii_uppercase)
g=itertools.izip(g1,g2)
Это гарантирует, что результирующий объект также является генератором.
Если вы предпочитаете использовать второе, вот как вы можете это сделать
g1=([s]*3 for s in string.ascii_lowercase)
g2=([s]*3 for s in string.ascii_uppercase)
g=(x+y for x,y in itertools.izip(g1,g2))
Скажем, у вас есть g1 и g2:
g1 = [
[['a', 'a', 'a'], ['e', 'e'], ['f', 'g']],
[['b', 'b', 'b'], ['e', 'e'], ['f', 'g']],
[['c', 'c', 'c'], ['e', 'e'], ['f', 'g']],
]
g2 = [
[[1, 1, 1], ['t', 'q'], ['h', 't']],
[[2, 2, 2], ['r', 'a'], ['l', 'o']],
[[3, 3, 3], ['x', 'w'], ['z', 'p']],
]
Чтобы получить это:
[a, a, a],[1, 1, 1]
[b, b, b],[2, 2, 2]
[c, c, c],[3, 3, 3]
Вы можете сделать это:
result1 = map(lambda a, b: (a[0], b[0]) , g1, g2)
# Which is like this :
[(['a', 'a', 'a'], [1, 1, 1]),
(['b', 'b', 'b'], [2, 2, 2]),
(['c', 'c', 'c'], [3, 3, 3])]
И для второго:
[a, a, a, 1, 1, 1]
[b, b, b, 2, 2, 2]
[c, c, c, 3, 3, 3]
result2 = map(lambda a, b: a[0]+b[0] , g1, g2)
# Which is like that :
[['a', 'a', 'a', 1, 1, 1],
['b', 'b', 'b', 2, 2, 2],
['c', 'c', 'c', 3, 3, 3]]