一、基础操作
列表是Python中一种重要的数据类型,常用于存储一组数据。在实际应用中,我们常常需要对列表进行分组操作。对于长度相同的数据,可以通过列表推导式实现分组。例如,对于列表[‘a’, ‘b’, ‘c’, ‘d’, ‘e’, ‘f’],可以按照长度为2的元素分为[[‘a’, ‘b’], [‘c’, ‘d’], [‘e’, ‘f’]]。
>>> data = ['a', 'b', 'c', 'd', 'e', 'f']
>>> N = 2
>>> [data[i:i+N] for i in range(0, len(data), N)]
[['a', 'b'], ['c', 'd'], ['e', 'f']]
上述代码中,range(0, len(data), N)表示从0开始,步长为N的整数序列,i表示序列中的元素。然后,使用切片操作获取每组数据,最终得到按照长度为N分组的列表。
二、分组排序
除了按照长度分组,有时候需要按照某种规则对列表进行分组。例如,对于学生考试成绩,可以按照分数段进行分组。
>>> scores = [('A', 80), ('B', 90), ('C', 85), ('D', 70), ('E', 95)]
>>> bins = [60, 70, 80, 90, 100]
>>> groups = {f'{i}-{j}': [] for i, j in zip(bins[:-1], bins[1:])}
>>> groups['<60'] = []
>>> for name, score in scores:
... for i, bin in enumerate(bins):
... if score < bin:
... if i == 0:
... groups['<60'].append((name, score))
... else:
... groups[f'{bins[i-1]}-{bin}'].append((name, score))
... break
...
>>> groups
{'<60': [('D', 70)],
'60-70': [],
'70-80': [('A', 80)],
'80-90': [('C', 85), ('B', 90)],
'90-100': [('E', 95)]}
上述代码中,bins表示分数段,groups是一个字典,用于存储各个分数段的学生信息。遍历每个学生的成绩,找到其所属的分数段,然后将其添加到对应分组中。
三、多层分组
有时候需要将列表按照多个维度进行分组操作。例如,对于多个学生的考试成绩,可以先按照科目分组,再按照分数段进行分组。
>>> scores = [('math', 'A', 80), ('math', 'B', 90), ('math', 'C', 85),
... ('physics', 'D', 70), ('physics', 'E', 95), ('physics', 'F', 75)]
>>> subjects = set([x[0] for x in scores])
>>> bins = [60, 70, 80, 90, 100]
>>> groups = {s: {f'{i}-{j}': [] for i, j in zip(bins[:-1], bins[1:])} for s in subjects}
>>> for subject, name, score in scores:
... for i, bin in enumerate(bins):
... if score < bin:
... if i == 0:
... groups[subject]['<60'].append((name, score))
... else:
... groups[subject][f'{bins[i-1]}-{bin}'].append((name, score))
... break
...
>>> groups
{'math': {'<60': [], '60-70': [], '70-80': [('A', 80)], '80-90': [('C', 85), ('B', 90)], '90-100': []},
'physics': {'<60': [('D', 70)], '60-70': [], '70-80': [('F', 75)], '80-90': [], '90-100': [('E', 95)]}}}
上述代码中,subjects表示科目,groups是一个嵌套字典,用于存储每个科目的学生信息。遍历每个学生的科目和成绩,找到其所属分数段和科目,然后将其添加到对应分组中。
四、按照首字母分组
有时候需要将列表按照首字母进行分组。例如,将[‘apple’, ‘banana’, ‘cherry’, ‘date’]按照首字母分组为{‘a’: [‘apple’], ‘b’: [‘banana’], ‘c’: [‘cherry’], ‘d’: [‘date’]}。
>>> data = ['apple', 'banana', 'cherry', 'date']
>>> groups = {}
>>> for item in data:
... initial = item[0]
... if initial not in groups:
... groups[initial] = []
... groups[initial].append(item)
...
>>> groups
{'a': ['apple'], 'b': ['banana'], 'c': ['cherry'], 'd': ['date']}
上述代码中,遍历列表中的每个元素,取得其首字母,然后根据首字母将元素添加到对应分组中。groups是一个字典,用于存储各个分组。
五、结语
Python列表分组操作有很多种实现方式,可以根据实际需求选择不同的方法。例如,对于二维列表,可以使用zip()函数实现分组;对于字符串列表,可以根据首字母进行分组。不同的实现方法有不同的适用场景,在实践中需要灵活运用。
原创文章,作者:LDIHJ,如若转载,请注明出处:https://www.506064.com/n/315732.html