Skip to main content
 首页 » 编程设计

python之将来自 2 个字典的值组合成一个 np.array python

2025年05月04日29lovecherry

我有两个指令

a = {0:[1,2,3,4], 1:[5,6,7,8],...} 
b = {0:[4,3,2,1], 1:[8,7,6,5],...} 

我想为每个键值对创建一个np.array c,如下所示

c1 = array([[1,4],[2,3],[3,2],[4,1]]) 
c2 = array([[5,8],[6,7],[7,6],[8,5]]) 

我该怎么做?是否可以将 np.array 存储在 python 字典中,以便我可以创建单个字典 c 而不是多个数组

请您参考如下方法:

是的,您可以将 np.array 放入 Python 字典中。只需使用 dict comprehension和 zip a 和 b 的列表。

>>> a = {0:[1,2,3,4], 1:[5,6,7,8]} 
>>> b = {0:[4,3,2,1], 1:[8,7,6,5]} 
>>> c = {i: np.array(list(zip(a[i], b[i]))) for i in set(a) & set(b)} 
>>> c 
{0: array([[1, 4], [2, 3], [3, 2], [4, 1]]), 
 1: array([[5, 8], [6, 7], [7, 6], [8, 5]])}