在Python中对嵌套列表进行排序和分组
我具有以下数据结构(列表列表)
[
['4', '21', '1', '14', '2008-10-24 15:42:58'],
['3', '22', '4', '2somename', '2008-10-24 15:22:03'],
['5', '21', '3', '19', '2008-10-24 15:45:45'],
['6', '21', '1', '1somename', '2008-10-24 15:45:49'],
['7', '22', '3', '2somename', '2008-10-24 15:45:51']
]
我希望能够
使用函数对列表重新排序,以便我可以按列表中的每个项目分组。 例如,我希望能够按第二列分组(以便所有21列在一起)
使用函数仅显示每个内部列表中的某些值。 例如,我想减少此列表,使其仅包含“ 2somename”的第四个字段值
所以列表看起来像这样
[
['3', '22', '4', '2somename', '2008-10-24 15:22:03'],
['7', '22', '3', '2somename', '2008-10-24 15:45:51']
]
m3clov3n asked 2020-08-10T09:14:34Z
8个解决方案
62 votes
对于第一个问题,您应该做的第一件事是使用运算符模块中的itemgetter按第二个字段对列表进行排序:
x = [
['4', '21', '1', '14', '2008-10-24 15:42:58'],
['3', '22', '4', '2somename', '2008-10-24 15:22:03'],
['5', '21', '3', '19', '2008-10-24 15:45:45'],
['6', '21', '1', '1somename', '2008-10-24 15:45:49'],
['7', '22', '3', '2somename', '2008-10-24 15:45:51']
]
from operator import itemgetter
x.sort(key=itemgetter(1))
然后,您可以使用itertools的groupby函数:
from itertools import groupby
y = groupby(x, itemgetter(1))
现在y是一个包含(元素,项目迭代器)元组的迭代器。 解释这些元组比显示代码更令人困惑:
for elt, items in groupby(x, itemgetter(1)):
print(elt, items)
for i in items:
print(i)
哪些打印:
21
[&#