在Python中,如何使用列表推导来遍历列表列表?

时间:2015-07-16 04:04:12

标签: python list list-comprehension

我有一个元组列表,其值和坐标为11点

dotted_array = [(0, 0, '.'), (2, 0, '.'), (3, 0, '.'), (0, 1, '.'), (2, 1, '.'), (0, 2, '.'), (2, 2, '.'), (3, 2, '.'), (0, 3, '.'), (2, 3, '.'), (3, 3, '.')]

我有5个列表的列表:

list_of_signs = [['+', '+', '-', '+', '+', '+', '+', '-', '+', '+', '-'], ['+', '+', '-', '+', '-', '+', '+', '-', '+', '+', '-'], ['+', '+', '-', '+', '+', '+', '+', '-', '+', '+', '-'], ['+', '-', '-', '+', '+', '+', '+', '-', '+', '+', '-'], ['+', '+', '-', '+', '+', '+', '+', '-', '+', '+', '-']]

该列表中的每个列表由i值+/-组成。这些+/-值对应于dotted_array中list元素的新值。

list_of_signs[] = [+,-,+,-,+.....11 values in each 'list_of_signs[]']

这是通过组合list_of_signs []的'value'和来自dotted_array []

的坐标的预期输出
coord_list = [[(0, 0, '+'), (2, 0, '+'), (3, 0, '-'), (0, 1, '+'), (2, 1, '+'), (0, 2, '+'), (2, 2, '+'), (3, 2, '-'), (0, 3, '+'), (2, 3, '+'), (3, 3, '-')], 4 More such lists ]

目前我:

coord_list= [(x[0],x[1],list_of_signs[0][0]) for x in dotted_array]

获得:

[(0, 0, '+'), (2, 0, '+'), (3, 0, '+'), (0, 1, '+'), (2, 1, '+'), (0, 2, '+'), (2, 2, '+'), (3, 2, '+'), (0, 3, '+'), (2, 3, '+'), (3, 3, '+')]

这个输出不仅错误,而且不一般。 如何对所有list_of_signs进行概括?

3 个答案:

答案 0 :(得分:5)

这是你想要的吗? -

>>> dotted_array = [(1,2,'.'), (4,5,'.'),(1,2,'.'), (4,5,'.'),(1,2,'.'), (4,5,'.')]
>>> list_of_signs = [['+','-','-','+','+','-'],['-','-','+','+','+','-']]


>>> coord_list = [[(x[0][0],x[0][1],x[1]) for x in zip(dotted_array,s)] for s in list_of_signs]


>>> coord_list
[[(1, 2, '+'), (4, 5, '-'), (1, 2, '-'), (4, 5, '+'), (1, 2, '+'), (4, 5, '-')], [(1, 2, '-'), (4, 5, '-'), (1, 2, '+'), (4, 5, '+'), (1, 2, '+'), (4, 5, '-')]]

zip函数将它在每个索引处作为参数接收的列表组合在一起,因此zip的返回列表(或迭代器)的索引将是第一个数组的第i个元素的元组,然后是第二个数组的第i个元素,所以上。

答案 1 :(得分:0)

让我知道它是否有效:

dotted_array = [(1,2,'.'), (4,5,'.')]
list_of_signs = [['+','-'],['-','-']]

coord_list = []
for idx1, list_of_sign in enumerate(list_of_signs):
    mylist = []
    for idx2, tup in enumerate(dotted_array):
        mylist += [(tup[0],tup[1],list_of_signs[idx1][idx2]) ]
    coord_list += mylist

print coord_list

将它写在一行会很有趣。

为两个数组和该数组的预期输出提供有效输入。

答案 2 :(得分:0)

dotted = zip(range(1,22,2),range(2,23,2),'.'*11)
signs = ['+-+-+-+-+-+']*5
print [(dotted[n][:-1]+(i,)) for s in signs for n,i in enumerate(s)]