如何在cython中切片列表

Question

我在 cython 中有一个列表，想在不使用 python 对象的情况下对其进行切片（为了速度）。

cdef int len = 100    
cdef int *q
cdef int *r

q = <int *>malloc( len *cython.sizeof(int) )

r = q[50:]

得到了这个错误：

r = q[50:]
    ^
------------------------------------------------------------

hello.pyx:24:9: Slicing is not currently supported for 'int *'.

有有效的方法吗？ “......目前不支持......”让我有点害怕。 我使用 cython 0.18

Answer 1

通过 Typed Memoryviews 可以实现快速切片和其他一些很酷的东西。但是为了进行切片，您需要一些关于数组的元数据，因此最好使用数组类型而不是普通指针。查找文档以获取更多信息：http://docs.cython.org/src/userguide/memoryviews.html。

你的问题的修改给出：

cdef int q_array[5] # c array
cdef int[:] q # 1D memview
cdef int[:] r # another 1D memview

q = q_array # point q to data
r = q[2:] # point r to a slice of q

r[0] = 5 # modify r

# test                                                                      
print q[2]
print r[0]

如果你真的不想要它，你仍然可以从切片创建指针：

# ...

cdef int* r_ptr
cdef int* q_ptr

r_ptr = &r[0]
q_ptr = &q[0]

print q_ptr[2]
print r_ptr[0]

也适用于 numpy 数组：

import numpy as np

cdef int[:] q = np.arange(100).astype('int32') # slow
cdef int[:] r

r = q[50:] # fast slicing

print r[0]