関数の入出力

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



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)



In [3]:

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



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]]



In [5]:

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









    



[1, 2, 3, 1, 2, 3, 1, 2, 3]