[tips][python] numpy

ブロードキャスト機能

配列

tipsです。メモなので日々追記していきます。

xには単一値でも配列でもよく、配列の場合は各要素に対して定めた演算を行い、和をとるとかしてくれるのがブロードキャスト機能。

import numpy as np
def function_2(x):
	return np.sum(x**2)

a = np.array([1, 2, 3])
res = function_2(a)
print("二乗和", res)

二乗和 14

• エラー

np.array() 関数による生成ではなく次のようにただのリストにすると、numpyがかた推論できずエラーになります。

c = [1, 2, 3]
res = function_2(c)
print("二乗和", res)

TypeError: unsupported operand type(s) for ** or pow(): 'list' and 'int'

参考