根据组号将数组数组排序为组

Question

我有一个数组。我试图将它们分成 4 组，每组 3 个。这是代码...

random.shuffle(top_twelve_players_info)
top_twelve_players_info_copy = top_twelve_players_info[:]

group_1 = []
group_2 = []
group_3 = []
group_4 = []

for i in range(3):
    group_1.append(top_twelve_players_info_copy[i])
    top_twelve_players_info_copy.remove(top_twelve_players_info_copy[i])
print(f"Group 1: {group_1}")

for i in range(3):
    group_2.append(top_twelve_players_info_copy[i])
    top_twelve_players_info_copy.remove(top_twelve_players_info_copy[i])
print(f"Group 2: {group_2}")

for i in range(3):
    group_3.append(top_twelve_players_info_copy[i])
    top_twelve_players_info_copy.remove(top_twelve_players_info_copy[i])
print(f"Group 3: {group_3}")

# print(top_twelve_players_info_copy)

for i in range(3):
    group_4.append(top_twelve_players_info_copy[i])
    # top_twelve_players_info_copy.remove(top_twelve_players_info_copy[i])
print(f"Group 4: {group_4}")

目前，top_twelve_players_info看起来像

[
    ['Krombopulos Michael', 10, 3],
    ['Scary Terry', 2, 11],
    ['Jerry Smith', 7, 6],
    ['Blim Blam ', 11, 2],
    ['Summer Smith', 1, 12],
    ['Tophat Jones', 5, 8],
    ['Beth Smith', 6, 7],
    ['Abradolf Lincler', 4, 9],
    ['Alan Rails', 12, 1],
    ['Morty Smith', 3, 10],
    ['Rick Sanchez', 9, 4],
    ['Xenon Bloom', 3, 10]
]

我一直在寻找更有效的实现，而不是重复 for 循环，甚至根据迭代次数以某种方式生成 group_1 或 group_2 变量。另外，出于某种原因，在当前的实现中，我只在制作第 4 组时才收到错误 "list index out of range"。所以我不得不评论删除行。这是为什么？

Answer 1

坚持使用 python 标准库（而不是 pandas，后者会提供更优雅的解决方案），您可能应该这样做：

from collections import defaultdict
groups = defaultdict(list)

players_cnt = 0
for group_no in range(4):
    for player_cnt in range(3):
        selected_player = top_twelve_players_info_copy[players_cnt]
        groups[group_no].append(selected_player)  # here defaultdict makes it more elegant - we do not need to check if group_no is a key in groups

现在，如果您真的需要，您可以提取组来分隔变量（但我强烈建议将其保存在由组号键入的字典中）：

group_1 = groups[1]  # ...etc.

Answer 2

你得到的错误是因为你在尝试按索引循环它时改变列表（从中删除项目）（所以当你的范围循环达到该值时索引 2 不再存在你的最后一个 for 循环）。

您可以通过使用范围循环对随机打乱的列表进行切片来简化该方法，其步长等于您要输出的组的大小。以下是一种将结果组放在一个列表中然后打印类似于您的示例的输出的方法。

import random

top_twelve_players_info = [['Krombopulos Michael', 10, 3], ['Scary Terry', 2, 11], ['Jerry Smith', 7, 6], ['Blim Blam ', 11, 2], ['Summer Smith', 1, 12], ['Tophat Jones', 5, 8], ['Beth Smith', 6, 7], ['Abradolf Lincler', 4, 9], ['Alan Rails', 12, 1], ['Morty Smith', 3, 10], ['Rick Sanchez', 9, 4], ['Xenon Bloom', 3, 10]]

random.shuffle(top_twelve_players_info)
groups = [top_twelve_players_info[i:i+3] for i in range(0, 12, 3)]

for i, group in enumerate(groups):
    print(f'Group {i + 1}: {group}')

# EXAMPLE OUTPUT (differs based on random shuffle results)
# Group 1: [['Beth Smith', 6, 7], ['Scary Terry', 2, 11], ['Krombopulos Michael', 10, 3]]
# Group 2: [['Xenon Bloom', 3, 10], ['Rick Sanchez', 9, 4], ['Tophat Jones', 5, 8]]
# Group 3: [['Morty Smith', 3, 10], ['Abradolf Lincler', 4, 9], ['Jerry Smith', 7, 6]]
# Group 4: [['Summer Smith', 1, 12], ['Alan Rails', 12, 1], ['Blim Blam ', 11, 2]]

或者如果您确实需要将每个组分配给自己的变量：

[group_1, group_2, group_3, group_4] = [top_twelve_players_info[i:i+3] for i in range(0, 12, 3)]

print(f'Group 1: {group_1}')
print(f'Group 2: {group_2}')
print(f'Group 3: {group_3}')
print(f'Group 4: {group_4}')

根据组号将数组数组排序为组

Sort array of arrays into groups based on group number

python

grouping

list

nested-lists

python-3.x