从 numpy 数组中，获取值列表的索引

Question

像这样：

>> arr = np.array([[0, 50], [100, 150], [200, 250]]) 
>>> values = [100, 200, 300] 

>>> arr in values

预计：

array([[False, False],
       [ True, False],
       [ True, False]])

结果：

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我写了下面的代码并且它有效，但是这段代码不能接受改变列表的长度

(arr==values[0]) | (arr==values[1]) | (arr==values[2])

Answer 1

例如将该变量的名称从 values 更改为与 valuess 不同的名称。

if valuess in arr.values :
print(valuess)

或

使用 lambda 函数。

假设您有一个数组：

nums = [0,1,5]
Check whether 5 is in nums:

(len(filter (lambda x : x == 5, nums)) > 0)

Answer 2

使用np.isin:

import numpy as np

arr = np.array([[0, 50], [100, 150], [200, 250]])
values = [100, 200, 300]

np.isin(arr, values)

结果：

array([[False, False],
       [ True, False],
       [ True, False]])

Answer 3

这应该有效，但仅适用于 2 级深度：

import numpy as np


def is_in(arr, values):
    agg = []
    for a in arr:
        if isinstance(a, list):
            result = is_in(a, values)
            agg.append(result)
        else:
            result = np.isin(arr, values)
            return result
    return agg

arr = np.array([[0, 50], [100, 150], [200, 250]])
values = [100, 200, 300]

print(is_in(arr, values))

从 numpy 数组中，获取值列表的索引

From numpy array, get indices of values list

python

numpy

numpy-ndarray