我有一個對象 -> {0: 0.8, 1: 0.2, 2: 0, 3: 0}我想將其轉換為 numpy 數組，其中鍵是索引，值是數組的值 -> [0.8, 0.2, 0, 0]最快、最有效的方法是什么？我正在使用 for 循環，但是有更好的方法嗎？
5 回答
慕妹3146593
TA貢獻1820條經驗 獲得超9個贊
keys，items并且values是從字典中獲取“內容”的最快方法。否則你必須迭代鍵。允許跳過索引和無序索引的通用方法：
In [81]: adict = {0: 0.8, 1: 0.2, 2: 0, 3: 0}
In [82]: keys = list(adict.keys())
In [83]: arr = np.zeros(max(keys)+1) # or set your own size
In [84]: arr[keys] = list(adict.values())
In [85]: arr
Out[85]: array([0.8, 0.2, 0. , 0. ])
波斯汪
TA貢獻1811條經驗 獲得超4個贊
如果輸入字典未排序或缺少值的解決方案：
d = {1:1.1, 2:2.2, 0: 0.8, 4:4.4}
sz = 1+max(d.keys()) # or len(d) if you are sure there are no missing values
x = np.full(sz, np.nan)
x[list(d.keys())] = list(d.values())
x
#array([0.8, 1.1, 2.2, nan, 4.4])
慕萊塢森
TA貢獻1810條經驗 獲得超4個贊
字典本質上是無序的。您需要根據鍵對值進行排序（假設您的字典中沒有丟失的鍵）：
a = {0: 0.8, 1: 0.2, 2: 0, 3: 0}
np.array([i[1] for i in sorted(a.items(), key=lambda x:x[0])])
另一種方法是在 numpy 中排序：
b = np.array(list(a.items()))
b[b[:,0].argsort()][:,1]
輸出：
[0.8, 0.2, 0, 0]
慕尼黑的夜晚無繁華
TA貢獻1864條經驗 獲得超6個贊
上面的答案給出了 dict_values，但方向正確
正確的方法是：
d = {0: 0.8, 1: 0.2, 2: 0, 3: 0}
np_array = np.array(list(d.values()))
添加回答
舉報
0/150
提交
取消