Python NumPy square()函数详解：数组平方计算指南

Python的numpy.square()函数返回一个新的数组，数组元素的值为原数组元素的平方。原数组保持不变。

Python中的numpy.square()函数示例

它是一个实用函数，能快速获得矩阵元素的平方。我们来看一下numpy square()函数的整数、浮点数和复数类型数组元素的例子。

使用numpy库的square()函数来对整数数组进行平方操作。

import numpy as np

# 整数数组
array_2d = np.array([[1, 2, 3], [4, 5, 6]])

print(f'源数组:\n{array_2d}')

array_2d_square = np.square(array_2d)

print(f'平方后的数组:\n{array_2d_square}')

输出：

源数组:
[[1 2 3]
 [4 5 6]]
平方后的数组:
[[ 1  4  9]
 [16 25 36]]

2. numpy square() 函数处理浮点数数组

import numpy as np

array_2d_float = np.array([1.2, 2.3, 5])

print(f'源数组:\n{array_2d_float}')

array_2d_float_square = np.square(array_2d_float)

print(f'平方后的数组:\n{array_2d_float_square}')

输出：

源数组:
[1.2 2.3 5. ]
平方后的数组:
[ 1.44  5.29 25.  ]

注意浮点数组中的整数已被转换为浮点数。

3. 使用numpy的square()函数对复数数组进行平方运算。

arr = np.array([1 + 2j, 2 + 3j, 4])

print(f'源数组:\n{arr}')

arr_square = np.square(arr)

print(f'平方后的数组:\n{arr_square}')

输出：

源数组:
[1.+2.j 2.+3.j 4.+0.j]
平方后的数组:
[-3. +4.j -5.+12.j 16. +0.j]

这里将整数元素转换为复数。更多详情请参考：NumPy官方API文档。