Python中的numpy.iscomplex介绍

简介

在Python的NumPy包中，我们可以使用numpy.iscomplex函数判断一个数是不是一个复数。该函数将返回一个布尔值。

语法

numpy.iscomplex(x)

参数说明：

• x： 需要判断的数，可以是一个标量或者多维数组。

返回值

该函数将返回一个与x的形状相同的布尔数组，如果x是一个复数，则返回True，否则返回False。

示例

import numpy as np

a = 1 + 2j
b = np.array([1, 2, 3 + 4j])
c = np.array([[1 + 1j, 2], [3, 4]])

print(np.iscomplex(a))  # True
print(np.iscomplex(b))  # [False False True]
print(np.iscomplex(c))  # [[ True False] [False False]]

在这个示例中，我们可以看到：

• a是一个复数，因此np.iscomplex(a)返回True。
• b是一个一维数组，其中最后一个元素是一个复数，因此np.iscomplex(b)返回[False False True]。
• c是一个二维数组，其中第一个元素是一个复数，因此np.iscomplex(c)返回[[True False] [False False]]。

总结

numpy.iscomplex函数可以帮助我们判断一个数是不是一个复数，非常方便实用。