2013년 7월 31일 수요일

Numpy 함수를 Array 적용하기

파이썬에서는 기본적으로 map 이라든지.

list comprehension의 방법으로 함수를 list나 dictionary에 적용이 가능하다.


기존의 map함수를 이용해서도 모든 array에 적용이 가능하지만,

반환되는 형태가 array가 아니라 python list형태로 반환이 되므로

다시 , array형태로 변경이 필요하게 된다.


그래서 , numpy에서는 array가 기본데이터 형태이므로 바로 array에 적용하는 방법이 필요하게 된다.

그 역할을 하는 함수가 vectorize이다.

import numpy as np

Arr = np.ndarray((3,4))
def foo(x):
    return x * x 

foo = np.vectorize(foo)  # 함수를 적용.  array에 적용가능하도록 변경. 

result = foo(Arr)  # Arr는 array의 이름임. 

댓글 없음:

댓글 쓰기