Python numpy.square() 函数返回一个新数组,其元素值作为源数组元素的平方。源数组保持不变。
它是一个快速求矩阵元素平方的实用函数。让我们看一下带有整数、浮点和复数类型数组元素的 numpy square() 函数的示例。
import numpy as np
# ints
array_2d = np.array([[1, 2, 3], [4, 5, 6]])
print(f'Source Array:\n{array_2d}')
array_2d_square = np.square(array_2d)
print(f'Squared Array:\n{array_2d_square}')
Output:
Source Array:
[[1 2 3]
[4 5 6]]
Squared Array:
[[ 1 4 9]
[16 25 36]]
import numpy as np
array_2d_float = np.array([1.2, 2.3, 5])
print(f'Source Array:\n{array_2d_float}')
array_2d_float_square = np.square(array_2d_float)
print(f'Squared Array:\n{array_2d_float_square}')
Output:
Source Array:
[1.2 2.3 5. ]
Squared Array:
[ 1.44 5.29 25. ]
请注意,浮点数组中的整数已转换为浮点数.
arr = np.array([1 + 2j, 2 + 3j, 4])
print(f'Source Array:\n{arr}')
arr_square = np.square(arr)
print(f'Squared Array:\n{arr_square}')
Output:
Source Array:
[1.+2.j 2.+3.j 4.+0.j]
Squared Array:
[-3. +4.j -5.+12.j 16. +0.j]
这里整数元素被转换为复数。参考:API Doc
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)