Python numpy获取col2中的值，给定与col1匹配的值数组

如何提取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]

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

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

a） 如何在Col0值为1或4（匹配idx）的情况下获取Col1的值

预期结果=[11,14]

b） 如何在行索引为1,4（匹配idx）的情况下获取Col1的值

预期结果=[12,14]

第一部分：

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

---

---

第二部分：

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

---

---

对于第2部分，我认为[idx][：，1]可以工作。谢谢@Pygirl]@py_新手你给错索引了。第二部分。

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

array([12, 14])