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

Question

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

例如：

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

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

标签： numpy

Answer 1

用于np.in1d过滤出n1和n2来自nodes：

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