从字典中提取多个数组并用 numpy 堆叠它们的 Pythonic 方法

Question

我有一本字典，里面有数组。我想用 numpy 抓取这些数组并使用 np.dstack 堆叠它们，但我不知道该怎么做。字典格式如下所示，当然减去了 nxx。此外，我按序列显示 nx 个数字，虽然字典包含键 n1、n2 直到最高 nvalue，但它们实际上并不在序列中。

{'n0': masked_array, 'n1': masked_array, 'n2': masked_array, ..., 'nxx': masked_array}

每一对的数据显示为

'n0': masked_array(
  data=[[0.13530391454696655, 0.26814308762550354, 0.26814308762550354,
         ..., --, --, --],
        [0.21222707629203796, 0.22386012971401215, 0.19620446860790253,
         ..., --, --, --],
        [0.22701908648014069, 0.23818181455135345, 0.23777709901332855,
         ..., --, --, --],
        ...,
        [--, --, --, ..., --, --, --],
        [--, --, --, ..., --, --, --],
        [--, --, --, ..., --, --, --]],
  mask=[[False, False, False, ...,  True,  True,  True],
        [False, False, False, ...,  True,  True,  True],
        [False, False, False, ...,  True,  True,  True],
        ...,
        [ True,  True,  True, ...,  True,  True,  True],
        [ True,  True,  True, ...,  True,  True,  True],
        [ True,  True,  True, ...,  True,  True,  True]],
  fill_value=-inf,
  dtype=float32)

使用

指定每个实体的打印

print(arraydict['n0'])

提供

[[0.13530391454696655 0.26814308762550354 0.26814308762550354 ... -- --
  --]
 [0.21222707629203796 0.22386012971401215 0.19620446860790253 ... -- --
  --]
 [0.22701908648014069 0.23818181455135345 0.23777709901332855 ... -- --
  --]
 ...
 [-- -- -- ... -- -- --]
 [-- -- -- ... -- -- --]
 [-- -- -- ... -- -- --]]

因此，如果它们是单独的变量，我可以做类似的事情（遵循 n 顺序很重要）

arraystack = np.dstack((n0, n1, n2, n3, ..., nxx, ))

我一辈子都无法概念化如何使用 np.dstack 从字典中提取和堆叠每个从 n0 开始的

arraystack = np.dstack(([arraydict[key] for key in arraydict]))

似乎有效，但似乎无法保持 n0、n1、n2、n3 序列。

Answer 1

你可以试试

arraystack = np.dstack([arraydict['n'+str(i)] for i in range(len(arraydict))])

这将确保订单得以保留。

从字典中提取多个数组并用 numpy 堆叠它们的 Pythonic 方法

Pythonic method of pulling multiple arrays out of a dictionary and stacking them with numpy

python

arrays

dictionary

numpy

numpy-ndarray