numpy获取与col1匹配的数组中col2的值

Question

如何在Col1中提取Col0匹配numpy数组中的任何值的值。我有一个np数组A idx。获取数组A的Col1中Col0值为1或4的所有值

A = np.array([[1, 11], [2, 12], [3, 13], [4,14]])
idx = [1, 4]

我可以得到一个这样的值。但是我不知道如何获取数组idx

vals = A[np.where(A[:,0]==4),1]
vals = A[np.where(A[:,0]==4),4]

a)我怎么能得到Col1的值在a中Col0的值是1或4(匹配idx)。

预期结果= [11,14]

b)我如何在A中获得Col1的值，其中行索引为1,4(匹配idx)

预期结果= [12,14]

Answer 1

第一部分:

idx = [1, 4]
A[np.isin(A[:,0], idx), 1]

---

array([11, 14])

---

第2部分:

idx = [1, 3]
A[idx,1]

---

array([12, 14])