如何在Cython中对列表进行切片

Question

如何在Cython中对列表进行切片

3

我有一个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 *'.

有一种高效的方法来做吗？"... not currently supported..."让我有点担心。我使用的是cython 0.18。

- Jean-Francois Gallant

q 不是一个列表，而是一个本地数组。我猜你需要使用更低级别的东西来处理它们。（传递一个数组和起始+结束索引。） - millimoose

@millimoose，我看了一下文档中的memoryview内容，但是我无法将其与我的简单示例配合使用。我对Cython和C编程都很陌生。你说的“低级别”和“传递数组和起始+结束索引”是什么意思？谢谢。 - Jean-Francois Gallant

在C语言中，当你使用数组时，通常不仅使用数组本身，还会使用“起始”和“长度”参数来指示函数应该处理的数组部分。数组和这两个索引一起表示一个“切片”（如果你查看快速排序等示例代码，就会发现这一点）。话虽如此，对于我不是非常熟悉的Cython来说，这可能是过度设计了。 - millimoose

@millimoose，你能告诉我在哪里可以找到快速排序的示例代码吗？这正是我需要切片数组的原因，为了实现KD树的多个qsort（我已经用Python完成了，但需要将其转换为Cython以提高速度）。 - Jean-Francois Gallant

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- dmytro · Accepted Answer

通过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]