Cython C-array initialization

Danilo Horta picture Danilo Horta · Sep 22, 2014 · Viewed 15.7k times · Source

I would like to do

cdef int mom2calc[3]
mom2calc[0] = 1
mom2calc[1] = 2
mom2calc[2] = 3

in a more compact way. Something similar to

cdef int mom2calc[3] = [1, 2, 3]

which is an invalid Cython syntax.

Note:

cdef int* mom2calc = [1, 2, 3]

is not an option because I cannot (automatically) converted it to a memory view.

Answer

Veedrac picture Veedrac · Sep 23, 2014
cdef int mom2calc[3]
mom2calc[:] = [1, 2, 3]

This works on raw pointers (although it isn't bounds checked then), memory views and fixed-sized arrays. It only works in one dimension, but that's often enough:

cdef int mom2calc[3][3]
mom2calc[0][:] = [1, 2, 3]
mom2calc[1][:] = [4, 5, 6]
mom2calc[2][:] = [7, 8, 9]