함수를 사용하여 numpy 배열을 만드는 방법은 무엇입니까?

Question

np.fromfunction을 사용하여 함수를 기반으로 특정 크기의 배열을 만듭니다. 그것은 다음과 같습니다

TypeError: only integer arrays with one element can be converted to an index 

Answer 1

기능은 NumPy와 배열을 처리해야 :

import numpy as np 
test = [[1,0],[0,2]] 
f = lambda i, j: sum(test[i]) 
matrix = np.fromfunction(f, (len(test), len(test)), dtype=int) 

그러나, 나는 다음과 같은 오류가 발생합니다. 이 작업을 얻을 수있는 쉬운 방법은 다음과 같습니다

import numpy as np 
test = [[1,0],[0,2]] 
f = lambda i, j: sum(test[i]) 
matrix = np.fromfunction(np.vectorize(f), (len(test), len(test)), dtype=int) 

np.vectorize 올바르게 배열을 처리 할 F의 벡터화 버전을 반환합니다.