API · NumPy

Python Numpy.square() - Carré

La fonction Python numpy.square() calcule le carré d'un tableau numpy.

Sur cette page

La fonction Numpy.square() calcule le carré de chaque élément du tableau donné.

C’est l’opération inverse de la méthode Numpy.sqrt().

Syntaxe de numpy.square()

numpy.square(arr, out=None)

Paramètres

arr tableau d’entrée
out Si out est donné, le résultat sera stocké dans out. out devrait avoir la même forme que arr.

Revenir

Il retourne un tableau du carré de chaque élément du tableau d’entrée, même si out est donné.

Exemples de codes: numpy.square()

import numpy as np

arr = [1, 3, 5, 7]

arr_sq = np.square(arr)

print(arr_sq)

Production:

[ 1  9 25 49]

Exemples de codes: numpy.square() avec le paramètre out

import numpy as np

arr = [1, 3, 5, 7]
out_arr = np.zeros(4)

arr_sq = np.square(arr, out_arr)

print(out_arr)
print(arr_sq)

Production:

[ 1  9 25 49]
[ 1  9 25 49]

out_arr a la même forme que arr, et le carré de arr y est enregistré.

Et la méthode numpy.square() retourne également le tableau carré, comme indiqué ci-dessus.

Si out n’a pas la même forme que arr, il déclenche une ValueError.

import numpy as np

arr = [1, 3, 5, 7]
out_arr = np.zeros(3)

arr_sq = np.square(arr, out_arr)

print(out_arr)
print(arr_sq)

Production:

Traceback (most recent call last):
  File "C:\Test\test.py", line 6, in <module>
    arr_sq = np.square(arr, out_arr)
ValueError: operands could not be broadcast together with shapes (4,) (3,) 

Exemples de codes: numpy.square() avec des nombres négatifs

import numpy as np

arr = [-1, -3, -5, -7]

arr_sq = np.square(arr)

print(arr_sq)

Production:

[ 1  9 25 49]

Exemples de codes: numpy.square() avec des nombres complexes

import numpy as np

arr = [1 + 2j, -2 - 1j, 2 - 3j, -3 + 4j]

arr_sq = np.square(arr)

print(arr_sq)

Production:

[-3. +4.j  3. +4.j -5.-12.j -7.-24.j]