"将列表拆分为 n 组并使用 Python 分配组 ID"

Question

给定一个元组列表，每个元组是一对变量的值。

[(0.0, 0.0), (0.0, 0.4), (0.0, 0.8), (0.0, 1.2000000000000002), (0.0, 1.6), (0.4, 0.0), (0.4, 0.4)]

Answer 1

您可以使用more-itertools将列表分成 N 组。

new_group = [[tuple(pair) + (group_id,) for pair in group_of_pairs]
             for group_id, group_of_pairs in enumerate(more_itertools.divide(ngroup, pair_var))]

或者你也可以使用numpy.array_split或numpy.split 。 这在处理大数据时要快得多，但请注意，此方法要求所有值都是相同的 dtype。

Answer 2

如果没有组 ID，这可以简单地通过列表理解来完成，如下所示：

new_list = [[item[0], item[1]] for item in output]

print(new_list)```

output: `[[0.0, 0.0], [0.0, 0.4], [0.0, 0.8], [0.0, 1.2000000000000002], [0.0, 1.6], [0.4, 0.0], [0.4, 0.4]]`

Adding an int of the first element just to fill in the Group ID space for example:

`new_list = [[item[0], item[1], int(item[0])] for item in output]`

output: `[[0.0, 0.0, 0], [0.0, 0.4, 0], [0.0, 0.8, 0], [0.0, 1.2000000000000002, 0], [0.0, 1.6, 0], [0.4, 0.0, 0], [0.4, 0.4, 0]]`

That Group ID assignment could be tweaked to fit your needs in the comprehension without need for numpy, I'm guessing.