如何读取一个 Numpy 数组并从另一个 Numpy 数组中获取值

Question

我是 Numpy / python 的新手，因此非常感谢任何帮助。 X 中的每一行都有 2 个数组。我必须从X2 获取值并将其附加到新数组，具体取决于X 数组值是0 还是1。

步骤：

1. 查看X 中连续有多少个0 或1
2. 从X2中的索引获取值
3. 对于X 中的每个0，将X2 中的值添加到arr0
4. 对于X 中的每个1 值，将X2 中的值添加到arr1
5. 遍历X 中的所有行

到目前为止，我有这个，但我不确定下一步该去哪里：

import numpy as np

X = np.array([[0,1,1,1,1],
             [0,1,1,1,0]])

X2 = np.array([[0.2,0.5,0.5,0.5,0.5],
              [0.3,0.6,0.6,0.6,0.3]])

index =[]
arr0 =[]
arr1=[]

for i in X:
    getindex = np.take(probs, i) # get the value from index
    index.append(getindex)

Answer 1

你可以试试 Numpy 模块take_along_axislink

X = np.array([[0,1,1,1,1],
          [0,1,1,1,0]])

X2 = np.array([[0.2,0.5,0.5,0.5,0.5],
          [0.3,0.6,0.6,0.6,0.3]])

extrac_val=np.take_along_axis(X2, X[:], axis=1)

输出

[[0.2 0.5 0.5 0.5 0.5]
 [0.3 0.6 0.6 0.6 0.3]]