[2-1] NumPyのモジュールをインポートします。



In [1]:

    
import numpy as np

[2-2] xの2乗を計算する関数を定義します。



In [2]:

    
def square(x):
    y = x ** 2
    return y

[2-3] いくつかの値を関数に代入してみます。



In [3]:

    
print square(1), square(2), square(3)

[2-4] NumPyのarrayオブジェクトを作成する例です。



In [4]:

    
xs = np.arange(1,6)
print xs









    



[1 2 3 4 5]

[2-5] arrayオブジェクトを関数に代入すると、計算結果もarrayオブジェクトで戻ります。



In [5]:

    
print square(xs)









    



[ 1  4  9 16 25]