我有两个指令
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]])}
