numpy.square() en Python

La fonction numpy.square() de Python renvoie un nouveau tableau avec la valeur des éléments au carré des éléments du tableau source. Le tableau source reste inchangé.

Exemples de numpy.square() en Python

C'est une fonction utilitaire pour obtenir rapidement le carré des éléments de la matrice. Regardons les exemples de la fonction numpy square() avec des éléments de tableau de type entier, flottant et complexe.

1. tableau entier 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}')

Sortie :

Source Array:
[[1 2 3]
 [4 5 6]]
Squared Array:
[[ 1  4  9]
 [16 25 36]]

2. tableau à virgule flottante square() de numpy

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}')

Sortie :

Source Array:
[1.2 2.3 5. ]
Squared Array:
[ 1.44  5.29 25.  ]

Remarquez que l'entier dans le tableau à virgule flottante a été converti en nombre à virgule flottante.

3. tableau de nombres complexes dans numpy square()

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}')

Sortie :

Source Array:
[1.+2.j 2.+3.j 4.+0.j]
Squared Array:
[-3. +4.j -5.+12.j 16. +0.j]

Ici, l'élément entier est converti en nombre complexe. Référence : Document API.