Python 中嵌套列表的排序和分组

2024-05-31

我有以下数据结构(列表的列表)

[
 ['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']
]

我希望能够

  1. 使用函数对列表重新排序,以便我可以按列表中的每个项目进行分组。例如,我希望能够按第二列进行分组(以便所有 21 都在一起)

  2. 使用函数仅显示每个内部列表中的某些值。例如,我想减少此列表以仅包含“2somename”的第四个字段值

所以列表看起来像这样

[
     ['3', '22', '4', '2somename', '2008-10-24 15:22:03'], 
     ['7', '22', '3', '2somename', '2008-10-24 15:45:51']
]

对于第一个问题,您应该做的第一件事是使用第二个字段对列表进行排序项目获取者 https://docs.python.org/3/library/operator.html#operator.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 https://docs.python.org/3/library/itertools.html#itertools.groupby功能:

from itertools import groupby
y = groupby(x, itemgetter(1))

现在 y 是一个包含 (element, item iterator) 元组的迭代器。解释这些元组比显示代码更令人困惑:

for elt, items in groupby(x, itemgetter(1)):
    print(elt, items)
    for i in items:
        print(i)

哪个打印:

21 <itertools._grouper object at 0x511a0>
['4', '21', '1', '14', '2008-10-24 15:42:58']
['5', '21', '3', '19', '2008-10-24 15:45:45']
['6', '21', '1', '1somename', '2008-10-24 15:45:49']
22 <itertools._grouper object at 0x51170>
['3', '22', '4', '2somename', '2008-10-24 15:22:03']
['7', '22', '3', '2somename', '2008-10-24 15:45:51']

对于第二部分,您应该使用此处已经提到的列表理解:

from pprint import pprint as pp
pp([y for y in x if y[3] == '2somename'])

哪个打印:

[['3', '22', '4', '2somename', '2008-10-24 15:22:03'],
 ['7', '22', '3', '2somename', '2008-10-24 15:45:51']]
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

Python 中嵌套列表的排序和分组 的相关文章

随机推荐