Why is numpy.dtype('float64') special?

后端 未结 2 1266
情深已故
情深已故 2021-01-05 01:48

Can someone explain the logic behind the output of the following script?

import numpy
if(numpy.dtype(numpy.float64) == None):
    print \"Surprise!!!!\"


        
相关标签:
2条回答
  • 2021-01-05 02:08

    Looks like an unfortunate accident: someone decided that dtype(None) would "default" to float (though dtype() is an error). Then someone else wrote dtype.__eq__ such that it converts its second argument to a dtype before comparing. So dtype(float) == None is dtype(float) == dtype(None) which is true.

    You can see a comment in the source code here: descriptor.c#L1217

    • Get typenum from an object -- None goes to NPY_DEFAULT_TYPE

    And of course NPY_DEFAULT_TYPE is float (at least usually).

    As for the __eq__ operator, it's here: descriptor.c#L3317. It does what I outlined:

    if (!PyArray_DescrCheck(other)) {
        if (PyArray_DescrConverter(other, &new) == NPY_FAIL) {
            return NULL;
        }
    }
    

    So that's a conversion from whatever is on the right-hand side of == to a dtype object, via the converter function mentioned before, which turns None into dtype(float).

    Edit: I found this pretty interesting and it seems like an accident, so I created a patch and submitted to the maintainers: https://github.com/numpy/numpy/pull/4532 .

    0 讨论(0)
  • 2021-01-05 02:23

    if you want to compare an arbitrary object against exactly None in python you need to use:

    object is None
    

    Like in this case any object may override its comparison operator to not do what you are expecting.

    As for why, dtype('float64') is equivalent to None in the context of dtypes in the same way dtypes are equivalent to typestrings

    np.dtype('i4') == 'i4'
    True
    

    Equality is not identity.

    As for why dtype(None) == dtype('float64'), many functions in numpy have dtype=None keyword arguments. In most cases this means default dtype which is dtype(None). An example is np.zeros. But there are exceptions, e.g. when the dtype can be inferred from the arguments, like in the case of np.arange(10) where the default dtype will be of integer type (np.intp I think).

    0 讨论(0)
提交回复
热议问题