Python检查数组的dtype - 浮点数或复数

Question

如果是 float 或 complex dtype，我如何检查 numpy 数组？对于简单的示例，以下检查都可以正常工作。

# these are True
a = np.zeros(10)
a.dtype == float
a.dtype == np.float
a.dtype == np.float64

b = np.zeros(10,dtype=complex)
b.dtype == complex
b.dtype == np.complex
b.dtype == np.complex128

但是，我有一个 dtype 数组 dtype('>f8')。之前的比较都没有将其识别为浮点数组。据我所知，字节顺序（> vs.

Answer 1

你试过numpy.isrealobj()和np.iscomplexobj()吗？

你的例子：

import numpy as np

a = np.zeros(10)
print(np.isrealobj(a)) # -> True
print(np.iscomplexobj(a)) # -> False

b = np.zeros(10,dtype=complex)
print(np.isrealobj(b)) # -> False
print(np.iscomplexobj(b)) # -> True

c=np.zeros(10, dtype='>f8')
print(np.isrealobj(c)) # -> True
print(np.iscomplexobj(c)) # -> False

np.isrealobj(x) 的文档状态：

如果 x 不是复数类型或复数数组，则返回 True。

检查的是输入的类型，而不是值。所以即使输入 虚部为零，isrealobj 计算结果为 False 如果数据类型很复杂。

也可以按值检查：np.isreal和np.iscomplex。

这有帮助吗？

Answer 2

您可以将all() 函数与isinstance() 函数结合使用。

如果列表的所有元素都为真，all() 函数将返回 True，并且您可以使用 isinstance() 检查 numpy 数组中的每个数字是否都是浮点数。由于 isinstance() 将 '>f8' 视为浮点数，因此比较不会引发您遇到的问题。

这是它的样子：

b = np.zeros(10,dtype='>f8')
all(isinstance(x, float) for x in b)

---

输出：

True