Извлечь значение из массива 1x3 с учетом двух других значений

Question

У меня есть целые числа 1x3.Учитывая пару целых чисел, как я могу извлечь третье целое число.Каждое целое число в векторе numpy уникально, и пара не будет включать себя (например, 3 и 3).

, например:

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

Для данных n1, n2 и массива 1x3Я хочу среднее значение, 5.

Answer 1

>>> arr = np.array([3, 5, 7])
>>> n1 = 3
>>> n2 = 7
>>> arr[(arr != n1) & (arr != n2)][0]
5

Это работает через логическое маскирование.Сначала исключите значения, которые соответствуют двум известным значениям.Поскольку вы знаете, что останется только одно значение, просто возьмите первый элемент.

Answer 2

Используйте np.in1d для фильтрации n1 и n2 от nodes:

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