迭代地收集矩阵的每个向量的前两个元素

时间:2013-12-28 14:16:09

标签: python list matrix

我有一个矩阵:

matrix = [['F', 'B', 'F', 'A', 'C', 'F'],
          ['D', 'E', 'B', 'E', 'B', 'E'],
          ['F', 'A', 'D', 'B', 'F', 'B'],
          ['B', 'E', 'F', 'B', 'D', 'D']]

我想删除并收集每个子列表的前两个元素,并将它们添加到新列表中。

到目前为止,我得到了:

while messagecypher:
    for vector in messagecypher:
        final.extend(vector[:2])

问题是; slice似乎没有删除元素,我最终得到了一个巨大的重复字符列表。我可以使用.pop(0)两次,但那不是很干净。

注意:我删除元素的原因是因为我需要继续遍历每个向量,直到矩阵为空

3 个答案:

答案 0 :(得分:2)

您可以保留slice并执行:

final = []
for i in range(len(matrix)):
    matrix[i], final = matrix[i][:2], final + matrix[i][2:]

请注意,这会同时将切片列表分配回matrix,并将切片的部分添加到final

答案 1 :(得分:1)

可以使用列表理解来完成任务,但这可能违反直觉:

>>> matrix = [['F', 'B', 'F', 'A', 'C', 'F'],
          ['D', 'E', 'B', 'E', 'B', 'E'],
          ['F', 'A', 'D', 'B', 'F', 'B'],
          ['B', 'E', 'F', 'B', 'D', 'D']]
>>> while [] not in matrix: print([i for var in matrix for i in [var.pop(0), var.pop(0)]])
['F', 'B', 'D', 'E', 'F', 'A', 'B', 'E']
['F', 'A', 'B', 'E', 'D', 'B', 'F', 'B']
['C', 'F', 'B', 'E', 'F', 'B', 'D', 'D']

编辑:

使用range使语法看起来更清晰:

>>> matrix = [['C', 'B', 'B', 'D', 'F', 'B'], ['D', 'B', 'B', 'A', 'B', 'A'], ['B', 'D', 'E', 'F', 'C', 'B'], ['B', 'A', 'C', 'B', 'E', 'F']] 
>>> while [] not in matrix: print([var.pop(0) for var in matrix for i in range(2)])
['C', 'B', 'D', 'B', 'B', 'D', 'B', 'A']
['B', 'D', 'B', 'A', 'E', 'F', 'C', 'B']
['F', 'B', 'B', 'A', 'C', 'B', 'E', 'F']

答案 2 :(得分:0)

删除元素不是完成任务的有效方式。它需要Python执行大量不必要的工作,以改变已删除元素留下的漏洞。相反,每次通过循环将切片移动两个位置:

final = []
for i in xrange(0, len(messagecypher[0]), 2):
    for vector in messagecypher:
        final.extend(vector[i:i+2])