Как инициализировать массивы numpy с фиксированными размерами в Cython?

Question

Как сделать пустые массивы numpy типа int в Cython? Следующие работы для меня для двойных или поплавковых массивов:

# make array of size N of type float 
cdef np.ndarray[float, ndim=1] myarr = np.empty(N) 
# make array of size N of type int 
cdef np.ndarray[int, ndim=1] myarr = np.empty(N) 

Однако, если я пытаюсь сделать то же самое с междунар, он не:

# this fails 
cdef np.ndarray[np.int, ndim=1] myarr = np.empty(N) 
# wanted to set first element to be an int 
myarr[0] = 5 

дает ошибку:

ValueError : Несоответствие типа буфера, ожидаемое 'int', но получившее 'double'

Поскольку, видимо, np.empty() возвращает двойной. Я пробовал:

cdef np.ndarray[np.int, ndim=1] myarr = np.empty(N, dtype=int) 

, но он дает ту же ошибку. Как это может быть сделано?

Answer 1

Включить заявление

cimport numpy as np 

и объявить массив как, скажем, np.int32_t:

cdef np.ndarray[np.int32_t, ndim=1] myarr = np.empty(N, dtype=np.int32) 

Вы можете роняйте 32 из деклараций типа и использовать

cdef np.ndarray[np.int_t, ndim=1] myarr = np.empty(N, dtype=np.int) 

но я предпочитаю быть явным о размере элементов в numpy arr ау.

Обратите внимание, что я также добавил dtype к empty; стандартным значением dtype empty является np.float64.

Answer 2

Wierd! У меня такая же ошибка, когда я пытался. Однако, глядя на сообщение об ошибке, я просто изменил область создания массива на функцию, и он компилируется! Я не знаю, почему это происходит, но.

import numpy as np 
cimport numpy as np 

ctypedef np.int_t DTYPE_t 
DTYPE=np.int 

def new_array(): 
    cdef int length = 10 
    cdef np.ndarray[DTYPE_t, ndim=1] x = np.zeros([length], dtype=np.int) 
    return x 

x = new_array() 

http://docs.cython.org/src/userguide/language_basics.html#python-functions-vs-c-functions Я думаю, что имеет какую-то информацию, связанную с обзорным питоном/с/смешанным переменными.