Numpy 检查数组是否全为零

在NumPy中，可以使用numpy.all()函数检查数组的所有元素是否都为非零值。在本文中，我们将展示如何使用该函数来检查一个NumPy数组是否全部为零。

语法

numpy.all(a, axis=None, out=None, keepdims=False)

• a：表示输入的数组
• axis：表示需要检查的轴。如果为None，则返回任意维度上所有元素的逻辑与。
• out：表示输出的数组
• keepdims：如果为True，则保持输出的数组与输入的数组具有相同的维度数。

示例

我们将从创建一个随机数组开始，然后使用NumPy检查它是否全为零。

import numpy as np

# 创建一个随机数组
arr = np.random.rand(3, 4, 5)

# 使用NumPy检查数组是否全为零
result = np.all(arr == 0)

# 输出结果
print(result)

这将打印出结果：False，因为随机数组中肯定存在非零值。

# 创建一个全为零的数组
arr = np.zeros((3, 4, 5))

# 使用NumPy检查数组是否全为零
result = np.all(arr == 0)

# 输出结果
print(result)

这将打印出结果：True，因为这个数组中的所有值都是零。

总结

在NumPy中，可以使用numpy.all()函数检查一个数组是否全部为零。这个函数可以实现一维数组和多维数组的检查，非常方便。