python - 使用追加构建二维数组

Question

Python:我想读取一组数据

(category, value): (0, 1) (0, 2) (1, 3) (1, 4)

将数组作为

[[1, 2],[3, 4]]

由于类别的最大数量未知，我想使用“append”方法动态创建二维数组。

我写了一个示例代码:

data = []
data.append([])

data[0].append(1)
data[0].append(2)

try:
  print (data[1])
except IndexError:
  data.append([])
finally:
  data[1].append(3)
  data[1].append(4)

print(data)

但是，我知道代码真的很难看，因为我使用“print”来检查对 data[1] 的访问。

这个问题还有更漂亮的解决方案吗？

Answer 1

使用itertools.groupby:

import itertools

a = [(0, 1), (0, 2), (1, 3), (1, 4)]
g = itertools.groupby(a, key=lambda x: x[0])
g = [list(i[1]) for i in g]