python - 从列表中的列表创建矩阵

Question

我有以下列表:

sortedList = [['2', 'f'],['5', 'B'],['8', '6'],['32','@'],['43', 'G'],['76', '.'],['173', 'v'],['200', '.'],['265', 'y']]

我尝试从此列表创建一个 3x3 矩阵，以便输出如下:

matrix = [['f','B','6'],['@','G','.'],['v','.','y']]

本质上，我需要找到一个函数，它将获取每个列表的最后一个字符串并将其放入长度为 3 个字符串的列表中。

Answer 1

您可以使用list comprehension收集每个项目中的最后一个字符串:

In [67]: [item[-1] for item in sortedList]
Out[67]: ['f', 'B', '6', '@', 'G', '.', 'v', '.', 'y']

然后，要将这些元素分成 3 组收集，请使用 grouper recipe :

In [68]: zip(*[(item[-1] for item in sortedList)]*3)
Out[68]: [('f', 'B', '6'), ('@', 'G', '.'), ('v', '.', 'y')]

请注意，石斑鱼配方 zip(*[iterator]*3) 调用迭代器，因此列表理解被替换为 generator expression .