Python中的 numpy.square()

numpy.square() 是一个用于计算数组元素的平方的函数。它是 NumPy 库中的一部分，用于进行数值计算和数组操作。此函数接受一个数组作为输入，并返回一个新的数组，其中包含输入数组中的每个元素的平方。

用法

numpy.square(x, /, out=None, *, where=True, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj])

参数

• x：要计算平方的输入数组。
• out：用于存储计算结果的输出数组。默认情况下，返回一个新的数组。
• where：布尔数组，用于指定在哪些位置计算平方。
• casting：定义数据类型转换的规则。
• order：将数组视为一个 C 风格的连续数据还是一个 Fortran 风格的连续数据。
• dtype：输出数组应具有的数据类型。
• subok：如果为 True，则返回一个子类数组。

返回值

numpy.square() 函数返回一个数组，其中包含输入数组 x 中元素的平方。返回数组与输入数组具有相同的形状。

示例

以下示例展示了如何在 Python 中使用 numpy.square() 函数：

import numpy as np

x = np.array([1, 2, 3, 4, 5])
result = np.square(x)

print(result)

输出：

[ 1  4  9 16 25]

上述示例中，我们创建了一个包含 1 到 5 的数组，并使用 numpy.square() 计算了每个元素的平方。最后打印结果数组。

总结

numpy.square() 是计算数组元素平方的方便函数，可以用于处理数值计算、平方特定的数据、计算方差等场景。不仅可以处理一维数组，还可以处理多维数组。快速、简单、高效地计算数组元素的平方是 numpy.square() 函数的特点。