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

Question

我有两个指令

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 而不是多个数组

Answer 1

是的，您可以将 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]])}