从字典对象创建边缘列表

Question

我有一个字典对象：

{0: array([1, 2, 3]), 1: array([4,5,6])}

所需的输出是边缘列表：

[0, 1], [0, 2], [0, 3], [1, 4], [1, 5], [1, 6]

如何在numpy中执行此操作？

Answer 1

使用普通的list-comprehension ：

import numpy as np
d = {0: np.array([1, 2, 3]), 1: np.array([4,5,6])}

[[key, v] for key, vals in d.items() for v in vals.tolist()]
# [[0, 1], [0, 2], [0, 3], [1, 4], [1, 5], [1, 6]]

如果需要数组：

np.array([[key, v] for key, vals in d.items() for v in vals.tolist()])