Numpy.Array in Python list?

前端 未结 5 1429
[愿得一人]
[愿得一人] 2020-12-01 18:42

I\'ve got a list (used as a stack) of numpy arrays. Now I want to check if an array is already in the list. Had it been tuples for instance, I would simply have written some

5条回答
  •  借酒劲吻你
    2020-12-01 19:02

    What about this:

    a = array([1, 1])
    
    l = [np.array([1,1]), np.array([2,2])]
    list(map(lambda x: np.array_equal(x, a), l)
    
    [True, False]
    

提交回复
热议问题