Làm thế nào để __contains__ hoạt động cho ndarrays?

Question

>>> x = numpy.array([[1, 2], 
...     [3, 4], 
...     [5, 6]]) 
>>> [1, 7] in x 
True 
>>> [1, 2] in x 
True 
>>> [1, 6] in x 
True 
>>> [2, 6] in x 
True 
>>> [3, 6] in x 
True 
>>> [2, 3] in x 
False 
>>> [2, 1] in x 
False 
>>> [1, 2, 3] in x 
False 
>>> [1, 3, 5] in x 
False 

Tôi không biết làm thế nào __contains__ hoạt động cho ndarrays. Tôi không thể tìm thấy tài liệu liên quan khi tôi tìm kiếm nó. Làm thế nào nó hoạt động? Và liệu nó có được ghi chép ở bất cứ đâu không?

Answer 1

Tôi tìm thấy nguồn cho ndarray.__contains__, trong numpy/core/src/multiarray/sequence.c. Như một lời nhận xét ở các bang nguồn,

thing in x 

tương đương với

(x == thing).any() 

cho một ndarray x, không phụ thuộc vào kích thước của x và thing. Điều này chỉ có ý nghĩa khi thing là vô hướng; kết quả phát sóng khi thing không phải là vô hướng gây ra các kết quả lạ mà tôi quan sát được, cũng như các số lẻ như array([1, 2, 3]) in array(1) mà tôi không nghĩ là thử. Nguồn chính xác là

static int 
array_contains(PyArrayObject *self, PyObject *el) 
{ 
    /* equivalent to (self == el).any() */ 

    PyObject *res; 
    int ret; 

    res = PyArray_EnsureAnyArray(PyObject_RichCompare((PyObject *)self, 
                 el, Py_EQ)); 
    if (res == NULL) { 
     return -1; 
    } 
    ret = array_any_nonzero((PyArrayObject *)res); 
    Py_DECREF(res); 
    return ret; 
} 

Answer 2

Có vẻ như numpy 's __contains__ đang làm một cái gì đó như thế này cho một trường hợp 2-d:

def __contains__(self, item): 
    for row in self: 
     if any(item_value == row_value for item_value, row_value in zip(item, row)): 
      return True 
    return False 

[1,7] công trình vì các yếu tố thứ 0 của hàng đầu tiên phù hợp với nguyên tố thứ 0 của [1,7]. Tương tự với [1,2] v.v. Với [2,6], số 6 khớp với số 6 ở hàng cuối cùng. Với [2,3], không có yếu tố nào khớp với một hàng ở cùng một chỉ mục. [1, 2, 3] là tầm thường vì các hình dạng không khớp.

Xem this để biết thêm và cũng có thể this ticket.