How do I check if a numpy dtype is integral?

Question:

How do I check if a numpy dtype is integral? I tried:

issubclass(np.int64, numbers.Integral)

but it gives False.


Update: it now gives True.

Asked By: Neil G

||

Answers:

Do you mean line 17?

In [13]:

import numpy as np
A=np.array([1,2,3])
In [14]:

A.dtype
Out[14]:
dtype('int32')
In [15]:

isinstance(A, np.ndarray) #A is not an instance of int32, it is an instance of ndarray
Out[15]:
True
In [16]:

A.dtype==np.int32 #but its dtype is int32
Out[16]:
True
In [17]:

issubclass(np.int32, int) #and int32 is a subclass of int
Out[17]:
True
Answered By: CT Zhu

Note that np.int64 is not a dtype, it’s a Python type. If you have an actual dtype (accessed through the dtype field of an array), you can make use of the np.typecodes dict you discovered:

my_array.dtype.char in np.typecodes['AllInteger']

If you only have a type such as np.int64, you can first obtain a dtype that corresponds to the type, then query it as above:

>>> np.dtype(np.int64).char in np.typecodes['AllInteger']
True
Answered By: user4815162342

Numpy has a hierarchy of dtypes similar to a class hierarchy (the scalar types actually have a bona fide class hierarchy that mirrors the dtype hierarchy). You can use np.issubdtype(some_dtype, np.integer) to test if a dtype is an integer dtype. Note that like most dtype-consuming functions, np.issubdtype() will convert its arguments to dtypes, so anything that can make a dtype via the np.dtype() constructor can be used.

http://docs.scipy.org/doc/numpy/reference/arrays.dtypes.html#specifying-and-constructing-data-types

>>> import numpy as np
>>> np.issubdtype(np.int32, np.integer)
True
>>> np.issubdtype(np.float32, np.integer)
False
>>> np.issubdtype(np.complex64, np.integer)
False
>>> np.issubdtype(np.uint8, np.integer)
True
>>> np.issubdtype(np.bool, np.integer)
False
>>> np.issubdtype(np.void, np.integer)
False

In a future version of numpy, we will make sure that the scalar types are registered with the appropriate numbers ABCs.

Answered By: Robert Kern

Depending on the use case the ducktyping

import operator
int = operator.index(number)

is a good method in my opinion. Plus it needs nothing numpy specific.

The only disadvantage is that in some cases you would have to try/except it.

Answered By: seberg

Building off previous answers and comments, I have settled on using the type attribute of the dtype object with Python’s builtin issubclass() method and the numbers module:

import numbers
import numpy

assert issubclass(numpy.dtype('int32').type, numbers.Integral)
assert not issubclass(numpy.dtype('float32').type, numbers.Integral)
Answered By: Johann

Since this question was asked, NumPy has added the appropriate registration with numbers, so this works:

issubclass(np.int64, numbers.Integral)
issubclass(np.int64, numbers.Real)
issubclass(np.int64, numbers.Complex)

This is more elegant than diving down to a more esoteric NumPy interface.

To perform this check on a dtype instance, use its .type property:

issubclass(array.dtype.type, numbers.Integral)
issubclass(array.dtype.type, numbers.Real)
issubclass(array.dtype.type, numbers.Complex)
Answered By: Neil G
Categories: questions Tags: , , ,
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.