python 使用较小的二维数组来映射另一个较大的数组

Question

我有一个较小的数组:

A = np.array([2013, 2014, 2015])
Aval = np.array([14, 10, 35])

我有另一个数组:

A2 = np.array([2013, 2014, 2015, 2013, 2014, 2015, 2013, 2013, 2013])

我想创建 A2val 以便:

Arval = np.array([14, 10, 35, 14, 10, 35, 14, 14, 14])

所以，我尝试使用数组 A 中的值映射到 A2 的元素并生成 A2val 的扩展版本

Answer 1

您可以使用np.searchsorted创建映射索引，然后索引到 Aval 以从中选择元素，就像这样 -

Aval[np.searchsorted(A,A2)]

如果 A 尚未排序，我们需要使用可选参数 sorter，如下所示 -

Aval[np.searchsorted(A,A2,sorter=A.argsort())]