Comparing two numpy arrays and removing elements

前端 未结 3 1862
日久生厌
日久生厌 2020-12-22 04:08

I have been going through several solutions, but I am not able to find a solution I need.

I have two numpy arrays. Let\'s take a small example<

3条回答
  •  自闭症患者
    2020-12-22 05:13

    Simply pass the negated version of boolean array returned by np.in1d to array x:

    >>> x = np.array([1,2,3,4,5,6,7,8,9])
    >>> y = [3,4,5]
    >>> x[~np.in1d(x, y)]
    array([1, 2, 6, 7, 8, 9])
    

提交回复
热议问题