関数の入出力

関数にスカラー,ベクトル,行列の何を入力して,何を受け取るのか意識する.

入力xをa倍して返す.


In [1]:
import numpy as np

def func(a, x):
    y = a*x
    return y

スカラーを入力して,スカラーを出力


In [2]:
a = 2
x = 4
y = func(a, x)
print(y)


8

ベクトルを入力して,ベクトルを出力.


In [3]:
a = 3
x = np.array([1,2,3])
y = func(a, x)
print(y)


[3 6 9]

行列を入力して,行列を出力


In [4]:
a = 3
x = np.array([[1,2,3], [4,5,6]])
y = func(a, x)
print(y)


[[ 3  6  9]
 [12 15 18]]

numpyでベクトルを作らないとうまくいかない.


In [ ]:
a = 3
x = [1,2,3]
y = func(a, x)
print(y)

In [ ]:
a = 43
b = 2
c = a*b
print(c)