Python Numpy.square()-二乗

  1. numpy.square() の構文
  2. コード例: numpy.square()
  3. コード例: num.square()out` パラメータ
  4. コード例:負の数の numpy.square()
  5. コード例:複素数の numpy.square()

Numpy.square() 関数は、指定された配列内のすべての要素の二乗を計算します。

Numpy.sqrt() メソッドの逆演算です。

numpy.square() の構文

numpy.square(arr,
          out=None) 

パラメーター

arr 入力配列
out out が指定された場合、結果は out に格納されます。outarr と同じ形状でなければなりません。

戻り値

out が指定されていても、入力配列の各要素の二乗の配列を返します。

コード例: numpy.square()

import numpy as np

arr = [1, 3, 5, 7]

arr_sq = np.square(arr)

print(arr_sq)

出力:

[ 1  9 25 49]

コード例: num.square()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)

出力:

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

out_arr の形状は arr と同じで、arr の二乗が格納されます。

そして、numpy.square() メソッドは上記のように二乗配列も返します。

outarr と同じ形状でない場合、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)

出力:

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

コード例:負の数の numpy.square()

import numpy as np

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

arr_sq = np.square(arr)

print(arr_sq)

出力:

[ 1  9 25 49]

コード例:複素数の numpy.square()

import numpy as np

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

arr_sq = np.square(arr)

print(arr_sq)

出力:

[-3. +4.j  3. +4.j -5.-12.j -7.-24.j]
comments powered by Disqus