從 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

使用 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 2

這應該有效，但僅適用於 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))

Answer 3

例如，還將該變量的名稱從 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)