是否可以在不关心尺寸的情况下检查numpy dtype?

时间:2018-01-16 13:56:56

标签: python numpy types

我希望有一个numpy dtype的对象,只是检查它是否是int类型,如果它有nbytes = 32,64,那就不关心......

有没有办法快速简单地做到这一点?

例如:

val = np.array([[2], [3], [4]]).astype(np.int32)
print val.type
>>> int

2 个答案:

答案 0 :(得分:1)

您可以使用np.issubdtype()功能:

In [44]: np.issubdtype(val.dtype, np.int)
Out[44]: True

答案 1 :(得分:1)

对于任何整数类型,

val.dtype.kind都会为您提供i