numpy.square

numpy.square

numpy.square() 函数

numpy.square() 函数用于计算数组中各元素的平方。

语法

numpy.square(arr, out=None)

参数

• arr: 输入的数组
• out: 可选参数，用于指定输出的数组

返回值

返回计算后的数组。

示例

import numpy as np

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

print(result)

输出：

[ 1  4  9 16]

使用示例

示例一：一维数组

import numpy as np

arr = np.array([5, 6, 7])
result = np.square(arr)

print(result)

输出：

[25 36 49]

示例二：二维数组

import numpy as np

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

print(result)

输出：

[[ 4  9]
 [16 25]]

示例三：指定输出数组

import numpy as np

arr = np.array([3, 4, 5])
out = np.zeros(3)
np.square(arr, out=out)

print(out)

输出：

[ 9. 16. 25.]

总结

numpy.square() 函数是一个方便的函数，用于计算数组中各元素的平方。你可以在一维数组、二维数组甚至更高维数组上使用该函数，还可以通过指定输出数组来对结果进行保存。