私は辞書オブジェクトを持っているのdictオブジェクト

Question

からedgelistを作成します。

{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：

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

：

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]] 

あなたは配列を必要とする場合