Python numpy.square() 函数返回一个新的数组,元素值作为源数组元素的平方。
Python numpy.square() 例子
它是一个实用函数,可以快速获取矩阵元素的平方. 让我们看看用整数,浮动和复杂类型数组元素的 numpy square() 函数的例子。
numpy 方块() int 数组
1import numpy as np
2
3# ints
4array_2d = np.array([[1, 2, 3], [4, 5, 6]])
5
6print(f'Source Array:\n{array_2d}')
7
8array_2d_square = np.square(array_2d)
9
10print(f'Squared Array:\n{array_2d_square}')
输出:
1Source Array:
2[[1 2 3]
3 [4 5 6]]
4Squared Array:
5[[ 1 4 9]
6 [16 25 36]]
numpy square(浮点数列)
1import numpy as np
2
3array_2d_float = np.array([1.2, 2.3, 5])
4
5print(f'Source Array:\n{array_2d_float}')
6
7array_2d_float_square = np.square(array_2d_float)
8
9print(f'Squared Array:\n{array_2d_float_square}')
输出:
1Source Array:
2[1.2 2.3 5. ]
3Squared Array:
4[ 1.44 5.29 25. ]
请注意,浮点数组中的整数已转换为 [浮点数]( / 社区 / 教程 / Python-float)。
numpy square() 复杂数组
1arr = np.array([1 + 2j, 2 + 3j, 4])
2
3print(f'Source Array:\n{arr}')
4
5arr_square = np.square(arr)
6
7print(f'Squared Array:\n{arr_square}')
输出:
1Source Array:
2[1.+2.j 2.+3.j 4.+0.j]
3Squared Array:
4[-3. +4.j -5.+12.j 16. +0.j]