cython 中操作字符串类型的内存视图

最新推荐文章于 2025-10-25 09:01:22 发布

原创最新推荐文章于 2025-10-25 09:01:22 发布 · 362 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #cython 字符串内存视图

Python 专栏收录该内容

9 篇文章

订阅专栏

该博客介绍了如何使用Cython和NumPy进行数组操作，特别是将字符串数组转换为字节数组，并创建视图。通过@cython.boundscheck(False)和@cython.wraparound(False)提升效率，`get_bytes_arr_view`函数展示了如何补零并获取字节视图，用于遍历和打印字符串数组元素。


cimport cython
cimport numpy as np
import numpy as np
from cython import address
import pandas as pd


@cython.boundscheck(False)
@cython.wraparound(False)
cpdef unsigned char[:,:] get_bytes_arr_view(np.ndarray bytes_arr):
    # 在每个bytes元素最后补零(可自行测试不补零的后果)
    bytes_arr = bytes_arr.astype(str("S%s" % (bytes_arr.dtype.itemsize+1)))
    # 按uint8获取视图
    cdef unsigned char[:,:] bytes_arr_view = bytes_arr.view('uint8').reshape(bytes_arr.size, bytes_arr.dtype.itemsize)
    return bytes_arr_view


def use_case():
    str_arr = np.array(["19:30:11", "19:30:31"])
    cdef np.ndarray time_arr = str_arr.astype(bytes)
    cdef unsigned char[:,:] time_arr_view = get_bytes_arr_view(time_arr)
    cdef unsigned char* time = b""
    # 遍历打印字符串array的元素
    for i in range(time_arr.shape[0]):
        time = address(time_arr_view[i, 0])
        print(time)