使用frompyfunc創建的python通用函數的數據類型

Question

關於numpy中的通用函數 ，我有以下問題。我怎樣才能定義一個通用函數，返回 與numpy內建函數相同類型的numpy數組。 下面的示例代碼：

import numpy as np 
def mysimplefunc(a): 
    return np.sin(a) 
mysimpleufunc = np.frompyfunc(mysimplefunc,1,1) 
a = np.linspace(0.0,1.0,3) 
print(np.sin(a).dtype) 
print(mysimpleufunc(a).dtype) 

結果輸出：

float64 
object 

任何幫助是非常讚賞:)

PS：我使用Python 3.4

Answer 1

我找到解決方案 （另請參閱discussion on stackoverflow）： 使用vectorize而不是frompyfunc

import numpy as np 
def mysimplefunc(a): 
    return np.sin(a) 
mysimpleufunc = np.vectorize(mysimplefunc) 
a = np.linspace(0.0,1.0,3) 
print(np.sin(a).dtype) 
print(mysimpleufunc(a).dtype)