numpy04 : 함수와 배열

배열을 함수의 인수로 사용

import numpy as np

def my_func(x):
    y = x * 2 + 1
    return y

a = np.array([[0, 1, 2],
              [3, 4, 5]])   #2차원 배열

b = my_func(a)  #인수로서 배열을 전달, 반환값으로서 배열을 받는다.

print(b)


# [[ 1  3  5]
#  [ 7  9 11]]