numpy - 给定其他两个值，从 1x3 numpy 数组中提取值

Question

我有一个 1x3 numpy 整数向量。给定一对整数，如何提取第三个整数。 numpy 向量中的每个整数都是唯一的，并且该对不会包含其自身(例如 3 和 3)。

例如:

>>> nodes
array([3, 5, 7], dtype=int16)
>>> n1 = 3
>>> n2 = 7

给定 n1、n2 和 1x3 数组，我想要中间值 5。

Answer 1

使用np.in1d从节点中过滤掉n1和n2:

nodes[~np.in1d(nodes, [n1, n2])]
# array([5])