Cython: ndarray字符串的内存视图（或直接使用ndarray索引）

Question

Cython: ndarray字符串的内存视图（或直接使用ndarray索引）

6

如何指定包含字符串的ndarray的内存视图？

char [:], char* [:]等方法均不可用。

为了说明我的问题，我的函数abc(...)的定义如下：

cdef void abc(char[:] in_buffer):
    cdef char * element
    element = address(in_buffer[1])
    ...

def main():
    cdef Py_ssize_t i, n = 100

    a = np.array(['ABC', 'D', 'EFGHI'])
    for i in range(n):
        abc(a)

如果无法使用memoryview，我是否可以自己实现直接数组访问？我需要在函数abc(...)中避免GIL。

编辑1：回复Bi Rico的答案。

我的目标是为了函数abc(...)释放GIL，并在其中使用c字符串函数处理ndarray in_buffer的字符串元素。也就是像下面这样做：

cdef void abc(char[:, ::1] in_buffer) nogil:
    cdef int max_elt_length = in_buffer.shape[1]+1
    cdef char element[max_elt_length+1]
    cdef int length

    for i in range(in_buffer.shape[0]+1):  # is this equivalent to in_buffer.dtype.itemsize + 1 ?
       element[max_elt_length] = 0   # add null-terminator for full-size elements
       memcpy(element, address(buffer[i, 0]), max_length)
       length = strlen(element)
       ...

- ARF

1个回答

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

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

- Bi Rico · Accepted Answer

问题在于numpy数组的数据类型必须具有固定的大小。当您创建一个“字符串”数组时，实际上是创建了一个固定长度的字符数组的数组。请尝试以下方法：

import numpy as np

array = np.array(["cat", "in", "a", "hat"])
array[2] = "Seuss"
print(array)
# ['cat' 'in' 'Seu' 'hat']
print(array.dtype)
# dtype('|S3')
print(array.dtype.itemsize)
# 3

考虑到这一点，您可以尝试像这样的操作：

cdef void abc(char[:, ::1] in_buffer):
    cdef char * element
    element = address(in_buffer[1, 0])

当您将数组传递给 abc 时，您需要执行以下操作：

a = np.array(['ABC', 'D', 'EFGHI'])
array_view = a.view('uint8').reshape(a.size, a.dtype.itemsize)
abc(array_view)

这只是一种方法，但在不了解你要做什么的情况下，我建议使用这种方法。