How does __contains__ work for ndarrays?

by

I found the source for ndarray.__contains__, in numpy/core/src/multiarray/sequence.c. As a comment in the source states,

thing in x

is equivalent to

(x == thing).any()

for an ndarray x, regardless of the dimensions of x and thing. This only makes sense when thing is a scalar; the results of broadcasting when thing isn’t a scalar cause the weird results I observed, as well as oddities like array([1, 2, 3]) in array(1) that I didn’t think to try. The exact source is

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

    int ret;
    PyObject *res, *any;

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

More Related Contents:

  1. What are the advantages of NumPy over regular Python lists?
  2. What is the difference between contiguous and non-contiguous arrays?
  3. Quick way to upsample numpy array by nearest neighbor tiling [duplicate]
  4. Automatically import modules when entering the python or ipython interpreter
  5. Plotting a fast Fourier transform in Python
  6. Cython: “fatal error: numpy/arrayobject.h: No such file or directory”
  7. How do I stack vectors of different lengths in NumPy?
  8. Cumsum reset at NaN
  9. Finding index of nearest point in numpy arrays of x and y coordinates
  10. Numpy meshgrid in 3D
  11. Vectorizing Haversine distance calculation in Python
  12. Sending live video frame over network in python opencv
  13. What is the difference between native int type and the numpy.int types?
  14. How to install numpy and scipy for Ironpython27?
  15. Scikit Learn SVC decision_function and predict
  16. How can I take the square root of -1 using python?
  17. Performance: Matlab vs Python
  18. Filtering a list based on a list of booleans
  19. Input and output numpy arrays to h5py
  20. Efficiently sorting a numpy array in descending order?
  21. Numpy: Divide each row by a vector element
  22. Running maximum of numpy array values
  23. One Hot Encoding using numpy [duplicate]
  24. Is there a numpy/scipy dot product, calculating only the diagonal entries of the result?
  25. numpy – evaluate function on a grid of points
  26. How can I draw lines into numpy arrays?
  27. Get the indices of N highest values in an ndarray
  28. Interweaving two NumPy arrays efficiently
  29. How to convert a 3d numpy array to 2d
  30. Numpy.dtype has the wrong size, try recompiling

Leave a Comment