Cythonはfloat ndarrayをint ndarrayに変換できません

Question

Cythonでnumpy ndarrayをある型から他の型にキャストすると、非常に奇妙なエラーが発生します。

とValueError：DTYPEのミスマッチをバッファし、期待される「私は非常に身近なメッセージを持っ

from convert_test import convert_to_int 
a_int=convert_to_int(np.zeros((2,2)).astype(np.float32)) 

：

cimport cython 
import numpy as np 
cimport numpy as np 

DTYPE = np.float32 
ctypedef np.float32_t DTYPE_t 

def convert_to_int(np.ndarray[DTYPE_t, ndim=2] float_matrix): 
    cdef np.ndarray[np.int, ndim=2] int_matrix=float_matrix.astype(np.int) 
    return int_matrix 

が、テストをコンパイルします。たとえば私は、次のconvert_test.pyxを書くときint object 'だが「long」を取得した

これはおそらく非常に明白ですが、私はまだそれを見ていません。

Answer 1

あなたはこれをやってみませんか？コンパイルとは、自分のコンピュータ上で正常に動作します：

cimport cython 
import numpy as np 
cimport numpy as np 

DTYPE = np.float32 
ctypedef np.float32_t DTYPE_t 

def convert_to_int(np.ndarray[DTYPE_t, ndim=2] float_matrix): 
    int_matrix = np.array(float_matrix, dtype=np.int32) 
    return int_matrix 

それとも、本当にmemoryviewをしたい場合：

def convert_to_int(DTYPE_t[:,:] float_matrix): 
    cdef int[:,:] int_matrix = np.array(float_matrix, dtype=np.int32) 
    return np.asarray(int_matrix) 

私はちょうどint_matrixとfloat_matrixのために近代的なtyped memoryviews構文に古いndarrayパラメータを変換された第2の回答。上記のスタイルは、行列の連続したメモリ書式[:,:]です（NumPyのデフォルト値はnumpy.zerosなどです）。

あなたが投稿した関数は、小数点以下のすべてを切り捨てる（切り上げない/切り捨てる）ので、意図したとおりであることに注意してください。

Answer 2

宣言で_tタイプを使用する必要があります：cdef np.ndarray [np.int_t、ndim = 2]。