以下是一个示例代码,用于按照数组的第一个值对数据进行分组:
def group_by_first_value(arr):
groups = {}
for item in arr:
key = item[0] # 获取数组的第一个值作为分组的键
if key in groups:
groups[key].append(item) # 如果分组已存在,则将数据添加到对应的分组列表中
else:
groups[key] = [item] # 如果分组不存在,则创建一个新的分组列表,并将数据添加进去
return groups
# 示例数据
arr = [[1, 'apple'], [2, 'banana'], [1, 'orange'], [3, 'pear'], [2, 'grape']]
# 按照数组的第一个值对数据进行分组
result = group_by_first_value(arr)
print(result)
输出结果为:
{1: [[1, 'apple'], [1, 'orange']], 2: [[2, 'banana'], [2, 'grape']], 3: [[3, 'pear']]}
以上代码通过遍历数组中的每个元素,将数组的第一个值作为分组的键,将对应的数据添加到分组列表中。如果分组已存在,则将数据添加到对应的分组列表中;如果分组不存在,则创建一个新的分组列表,并将数据添加进去。最后返回分组结果。