Question

请考虑以下代码：

arr = np.array([0, 1, 3, 5, 5, 6, 7, 9, 8, 9, 3, 2, 4, 6])
mapping = np.array([0, 10, 20, 30, 40, 55, 66, 70, 80, 90])
res = np.zeros_like(arr)
min_val = 0
max_val = 10

for val in range(min_val, max_val):
    res[arr == val] = mapping[val]

print(res)

Numpy数组arr可以多次出现间隔[min_val, max_val)中的整数。 mapping数组将具有每个整数的映射，并且mapping数组的大小将为max_val。 res数组是结果数组。

for循环将arr中多个出现的元素替换为mapping中的相应值。例如，0中的arr值将被mapping[0]替换，5中的arr被mapping[5]替换。

以上代码的结果如下。

[ 0 10 30 55 55 66 70 90 80 90 30 20 40 66]

问题：如何使用Numpy而不是for循环来执行此操作？

答案是使用Numpy的fancy indexing

Answer 1

您可以简单地将arr用作mapping的索引数组：

mapping[arr]

输出为

[ 0 10 30 55 55 66 70 90 80 90 30 20 40 66]

您可以阅读SciPy官方文档中的indexing arrays。文档中的示例：

>>> x = np.arange(10,1,-1)
>>> x
array([10,  9,  8,  7,  6,  5,  4,  3,  2])
>>> x[np.array([3, 3, 1, 8])]
array([7, 7, 9, 2])

索引数组的值用作源数组的索引。

multidimensional arrays也可以这样做：


>>> x = array([[ 0,  1,  2],
...            [ 3,  4,  5],
...            [ 6,  7,  8],
...            [ 9, 10, 11]])
>>> rows = np.array([[0, 0],
...                  [3, 3]], dtype=np.intp)
>>> columns = np.array([[0, 2],
...                     [0, 2]], dtype=np.intp)
>>> x[rows, columns]
array([[ 0,  2],
       [ 9, 11]])

Answer 2

只需使用private void Button_Click(object sender, RoutedEventArgs e) { CanvasViewModel.Instance.Switch(2); }来访问新的Numpy数组中的正确元素：

  <ion-item>
    <ion-label class="ion-text-wrap">Long Text</ion-label>
  </ion-item>

如果您希望将其作为mapping[arr]：

>>> arr = np.array([0, 1, 3, 5, 5, 6, 7, 9, 8, 9, 3, 2, 4, 6])
>>> mapping = np.array([0, 10, 20, 30, 40, 55, 66, 70, 80, 90])
>>> print(mapping[arr])
array([ 0, 10, 30, 55, 55, 66, 70, 90, 80, 90, 30, 20, 40, 66])

用相应的值替换Numpy数组中多个出现的元素的集合

2 个答案: