NumPy库学习之np.square函数

NumPy库学习之np.square函数

一、简介

np.square 是 NumPy 库中的一个函数，用于计算数组中每个元素的平方。此函数对输入数组的每个元素执行平方操作，返回一个新的数组，其中包含原始元素的平方值。

二、语法和参数

函数的基本语法如下：

numpy.square(x)

参数说明：

• x：输入数组或数值，要求是可以进行数学平方运算的数据类型。

三、实例

3.1 基础使用

import numpy as np

# 创建一个包含几个整数的数组
arr = np.array([1, 2, 3, 4])
# 计算数组中每个元素的平方
squared_arr = np.square(arr)

print("原始数组:", arr)
print("平方后的数组:", squared_arr)

输出：

原始数组: [1 2 3 4]
平方后的数组: [1 4 9 16]

3.2 处理浮点数数组

import numpy as np

# 创建一个包含浮点数的数组
float_arr = np.array([0.5, 2.5, 3.5])
# 计算数组中每个元素的平方
squared_float_arr = np.square(float_arr)

print("浮点数数组:", float_arr)
print("平方后的浮点数数组:", squared_float_arr)

输出：

浮点数数组: [0.5 2.5 3.5]
平方后的浮点数数组: [0.25 6.25 12.25]

四、注意事项

• np.square 函数可以处理多维数组，且会逐元素地进行平方运算。
• 输入数组 x 中的元素可以是整数、浮点数或复数，函数会相应地返回它们的平方。
• 此函数不会就地修改输入数组，而是返回一个新的数组。如果需要在原数组上进行修改，可以对原数组重新赋值。