我有一个ndarray,其中包含按特定顺序排列的字典键。我想创建另一个包含相应键值的ndarray。必须保持顺序。 一种明显的方法是逐个元素地迭代包含键的数组,但问题是无法事先知道数组的形状。
是否可以展平键的ndarray并对其进行迭代以生成平坦的ndarray值,并最终在不破坏顺序的情况下对其进行拆散?
.webp
N。 B .:在上面的示例中,图案数组是2D的,并非总是如此。另外,它可能不包含字典的所有键。
答案 0 :(得分:5)
您可以将np.vectorize
与dict.get
一起使用:
d = np.vectorize(mydict.get)
res = d(input_pattern)
print(res)
array([[1, 6],
[2, 5],
[3, 4]])
答案 1 :(得分:0)