在for循環中訪問itertools產品的元素

Question

我有一個列表，它是從itertools產品追加其他一些結果的結果列表。我想要的是能夠使用for循環訪問列表列表中的列表中的每個元素，但我無法訪問所有元素。我只能訪問最後一個列表的元素。結果是一個非常大的列表，如：[[1,2，]，[4,3,6,19] [2,0,34,9,9] ......... [n ]。我希望能夠訪問for循環中列表中每個列表中的所有元素，但只能獲取最後一個列表中的元素。 Itertools僅顯示內存中元素的位置。

myListofLists=[]        
from itertools import product 
animal = ['dog','cat'] 
number = [5,45,8,9] 
color = ['yellow','blue','black'] 
myProduct = product(animal, number, color): 
myListOfLists.append(myProduct) 
for item in myListOfList: 
    list(items) 

這使得結果是這樣的：[[ '狗'，5， '黃'] [ '狗'，45， '藍']，......] 這是隻顯示在shell中，但我試圖將結果寫入文件，我只能看到最後一個列表的元素。 （1）如何訪問鏈表中每個列表的所有元素，以及（2）除了向我顯示元素的位置外，我怎樣才能在python shell中查看結果記憶。謝謝！

Answer 1

讀您的文章很多次，我認爲你有興趣在此語法：

for animal, number, color in ...: 

它允許您訪問從內部列表中的元素。

import itertools 
animals = ['dog', 'cat'] 
numbers = [5, 45, 8, 9] 
colors = ['yellow', 'blue', 'black'] 
for animal, number, color in itertools.product(animals, numbers, colors): 
    print "Animal is {0}, number is {1}, and color is {2}".format(animal, number, color) 

查看運行上ideone

編輯：閱讀您的意見在其他線程，我猜你感興趣的迭代內的元組（？）。您不必將它們轉換爲列表，您可以簡單地遍歷它們。在ideone

EDIT 2

import itertools 
animals = ['dog', 'cat'] 
numbers = [5, 45, 8, 9] 
colors = ['yellow', 'blue', 'black'] 
for triplet in itertools.product(animals, numbers, colors): 
    for element in triplet: 
     print element 

演示：您可以嘗試爲自己的分數件事的做法。

import itertools 
animals = ['dog', 'cat'] 
numbers = [5, 45, 8, 9] 
colors = ['yellow', 'blue', 'black'] 
general_list = ['horse', 'blue', 'dog', 45] 

def give_score(element): 
    return 1 if element in general_list else 0 

for triplet in itertools.product(animals, numbers, colors): 
    print sum([give_score(element) for element in triplet]) 

演示here

Answer 2

itertools.product本身返回一個對象，你可以遍歷。例如：

>>> from itertools import product 
>>> animal = ['dog','cat'] 
>>> number = [5,45,8,9] 
>>> color = ['yellow','blue','black'] 
>>> myProduct = product(animal, number, color) 
>>> for p in myProduct: 
...  print p 
...  
('dog', 5, 'yellow') 
('dog', 5, 'blue') 
('dog', 5, 'black') 
[...] 
('cat', 9, 'blue') 
('cat', 9, 'black') 

在此之後，myProduct耗盡（出元素的產生），所以如果你再次循環，你不會得到任何東西：

>>> for p in myProduct: 
...  print p 
...  
>>> 

如果你想兌現myProduct成列表，你可以這樣做：

>>> myProduct = list(product(animal, number, color)) 
>>> len(myProduct) 
24 
>>> myProduct[7] 
('dog', 8, 'blue') 

通過追加product對象本身的列表，你沒有得到的列表迭代它會p結果，只有對象。

>>> myProduct = product(animal, number, color) 
>>> myProduct 
<itertools.product object at 0x8e3ce3c> 

如果你願意，你可以在一些地方保存該對象，並從中獲取內容的相同方法：

>>> some_list = [myProduct, myProduct] 
>>> next(some_list[0]) 
('dog', 5, 'yellow') 
>>> next(some_list[1]) 
('dog', 5, 'blue') 
>>> next(some_list[1]) 
('dog', 5, 'black') 
>>> next(some_list[0]) 
('dog', 45, 'yellow') 

但在這裏，我不認爲這是你在找什麼。