我想将数组arr_2的值分配给另一个数组arr_1。但是,我想基于2个选择标准来执行此操作。作为一个工作示例,我将选择标准定义为

import numpy as np

# An array of -1 values of shape(10,): [-1 -1 -1 -1 -1 -1 -1 -1 -1 -1]
arr_1 = np.zeros(10, dtype=int) - 1
# An array of 0-9 values of shape(10,): [0 1 2 3 4 5 6 7 8 9]
arr_2 = np.arange(10)

# Create an initial selection of values we want to change
# In this example: even indices: [ T F T F T F T F T F]
selection_a = np.arange(10) % 2 == 0

# Create a second selection based on selection_a: [F F F T T]
selection_b = arr_2[selection_a] > 5


基于这两个选择标准,我想将两个条件均满足的arr_2值分配给数组arr_1。即等同于[F F F F F F T F T F]

arr_1[selection_a][selection_b] = arr_2[selection_a][selection_b]


如果我在分配之前检查方程式的两边,它们会得出我期望的值:

print(arr_1[selection_a][selection_b]) # yields [-1 -1]
print(arr_2[selection_a][selection_b]) # yields [ 6, 8]


但是,分配本身不会分配值,即arr_1保持不变。我的问题是,为什么会这样?

注意:我知道在大多数情况下(甚至可能在所有情况下),可以通过创建一个标准来规避这一问题,但是我想知道为什么使用两个单独的标准将行不通。

如果有任何不清楚的地方,请让我知道,我会尽力澄清。

编辑

我对此进行了进一步研究,问题似乎在方程的左侧,类似于

arr_1[selection_a][selection_b] = 5


也不起作用。

最佳答案

单独使用[...]运算符会在数组中创建一个选择,您可以从中读取内容,也可以向其中写入内容。第二次使用[...]将毫无问题地读取值。但是就写作而言,第二次访问将是根据您的情况创建的,与选择规则匹配的临时数组(即形状为(2,)的临时副本,它是原始数据的副本)。

编辑:布尔索引是所谓的“高级索引”的一部分。将索引保持为简单片可避免这些复制问题。

关于python - 为什么numpy不允许分配双索引的numpy数组?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/52383664/

10-12 16:53
查看更多