通过C模块中的Python API访问数组的数组

在 C 语言中,我们可以使用 Python 的 C API 来访问和操作数组的数组(即二维数组或嵌套列表)。通常,我们可以使用 Python C API 提供的 PyListObject 和 PySequence 相关函数来访问 Python 传递过来的列表结构。

1、问题背景

在 Python 中创建了一个包含数组的数组,并将其传递给 C 模块。我们需要通过 C 模块中的 Python API 访问此数组的每个子数组。

2、解决方案

为了访问传递给 C 模块的数组的数组,可以使用以下步骤:

在 C 模块中,使用 PyArray_SimpleNewFromData() 函数创建一个新的 NumPy 数组。此函数将创建一个新的数组,并使用提供的数据填充它。

将传递给 C 模块的数组的数组的元素复制到新创建的数组中。这可以使用 PyArray_CopyInto() 函数来完成。

使用 PyArray_NDIM() 函数获取新创建的数组的维度数。

使用 PyArray_SHAPE() 函数获取新创建的数组的形状。

使用 PyArray_GETPTR1() 函数获取新创建的数组的数据指针。

使用数据指针访问新创建的数组中的元素。

以下是一个代码示例:

# C 代码

# 包含 NumPy 头文件

# conda install numpy

# pip install numpy

# apt-get install python3-numpy  (For Debian / Ubuntu)

# yum install python3-numpy  (For CentOS / Red Hat)

# dnf install python3-numpy  (For Fedora)

# zypper install python3-numpy  (For OpenSUSE)

# Include the necessary header files

# For Windows: Python.h must be placed in the project directory

# For Linux: python3.X/Include/Python.h

# For macOS: /Library/Frameworks/Python.framework/Versions/3.X/Headers/Python.h

# https://www.python.org/dev/embedding/

# https://www.oreilly.com/library/view/python-c-extension/0596001566/re314.html

# https://docs.python.org/3/c-api/array.html

# https://docs.scipy.org/doc/numpy/reference/c-api/c-api-intro.html

//https://stackoverflow.com/questions/5390264/numpy-array-of-objects-passing-to-c-and-access-to-these-objects-in-c

# https://docs.scipy.org/doc/numpy-1.15.1/reference/c-api/ndarray.html

# https://github.com/numpy/numpy/issues/11485

# https://github.com/numpy/numpy/issues/2662

# https://www.oreilly.com/library/view/python-c-extension/0596001566/re335.html

# https://stackoverflow.com/questions/47386916/how-to-create-a-numpy-array-of-objects-in-c-extension

//https://cs231n.github.io/python-numpy-tutorial/

# https://jonathannienaber.de/programming/python-native-memoryview-efficient-cpython-extension-types/

# https://numpy.org/doc/stable/c-api/c-api-reference.html

# https://numpy.org/doc/stable/reference/generated/numpy.ndarray.html

# https://numpy.org/doc/stable/user/basics.types.html

//https://stackoverflow.com/questions/37056848/how-to-pass-a-numpy-array-from-python-to-c

//https://numpy.org/doc/stable/reference/c-api/shape.html

//https://stackoverflow.com/questions/68689338/convert-numpy-array-of-bytes-into-list-of-strings

# https://numpy.org/doc/stable/reference/c-api/array.html

//https://numpy.org/doc/stable/user/basics.creation.html#creating-arrays

# https://numpy.org/doc/stable/reference/c-api/dtype.html

# https://numpy.org/doc/stable/reference/c-api/dtype_buffer.html

# https://numpy.org/doc/stable/reference/c-api/dtype_flag.html

# https://scipython.com/blog/interacting-between-python-and-c-data-structures/

# 定义必要的变量和数据结构

PyObject*PyArray,*PySubArray;

intPyNumDims=0;

npy_intp*PyShape=NULL;

PyArray_Descr*PyDtype=NULL;

void*PyDataPtr=NULL;

//将传递给C模块的数组的数组转换为兼容的NumPy数组

PyObject*PyArrayConvert(PyObject*PyArray) {

//检查输入数组是否为数组对象

if(!PyArray_Check(PyArray)) {

PyErr_SetString(PyExc_TypeError,"Input must be a NumPy array");

returnNULL;

  }

//检查输入数组是否是数组的数组

PyNumDims=PyArray_NDIM(PyArray);

if(PyNumDims!=2) {

PyErr_SetString(PyExc_TypeError,"Input must be a 2-dimensional array");

returnNULL;

  }

//检查输入数组的元素类型是否兼容

PyDtype=PyArray_DESCR(PyArray);

if(PyDtype->type_num!=NPY_OBJECT) {

PyErr_SetString(PyExc_TypeError,"Input array must have object data type");

returnNULL;

  }

//将输入数组转换为兼容的NumPy数组

PyArray=PyArray_FromArray(PyArray,PyArray_DescrFromType(NPY_OBJECT),NPY_ARRAY_DEFAULT);

if(PyArray==NULL) {

PyErr_SetString(PyExc_RuntimeError,"Failed to convert input array");

returnNULL;

  }

returnPyArray;

}

//从兼容的NumPy数组中提取数组的数组元素

PyObject**PyExtractSubArrays(PyObject*PyArray) {

//获取兼容的NumPy数组的形状和数据指针

PyShape=PyArray_SHAPE(PyArray);

PyDataPtr=PyArray_DATA(PyArray);

//分配内存来存储提取的数组的数组元素

PyObject**PySubArrays=malloc(PyShape[0]*sizeof(PyObject*));

if(PySubArrays==NULL) {

PyErr_SetString(PyExc_MemoryError,"Failed to allocate memory for sub-arrays");

returnNULL;

  }

//提取兼容的NumPy数组的数组的数组元素

for(inti=0;i<PyShape[0];i++) {

PySubArrays[i]=((PyObject**)PyDataPtr)[i];

  }

returnPySubArrays;

}

//释放提取的数组的数组元素的内存

voidPyFreeSubArrays(PyObject**PySubArrays) {

free(PySubArrays);

}

//在C模块中访问提取的数组的数组元素

voidPyAccessSubArrays(PyObject**PySubArrays) {

//循环访问提取的数组的数组元素

for(inti=0;i<PyShape[0];i++) {

//访问提取的数组的数组元素的元素

for(intj=0;j<PyShape[1];j++) {

//根据需要访问提取的数组的数组元素的元素的值

//printf("%d\n", ((int**)PySubArrays[i])[j]);

   }

  }

}

//在C模块中释放kompatiblerNumPy数组的内存

voidPyFreeArray(PyObject*PyArray) {

Py_DECREF(PyArray);

}

请注意,此代码假设传递给 C 模块的数组的数组是一个二维数组,并且它的元素是对象。如果数组的数组具有不同的维度或元素类型,则需要修改代码以支持不同的情况。

以下是一个 Python 代码示例,演示如何使用上述 C 代码访问数组的数组:

# Python 代码

# 导入必要的库

importnumpyasnp

importgeoms

# 创建一个包含数组的数组

A=np.empty((1,2),dtype=object)

A[0,0]=np.random.rand(3,3)

A[0,1]=np.random.rand(5,5)

# 将数组的数组传递给 C 模块

geoms.gm_unique_all(A)

通过这种方式,C 代码可以访问 Python 传递的二维数组(列表的列表),并在 C 端处理数据后返回新的 Python 结构。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 220,137评论 6 511
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,824评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 166,465评论 0 357
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,131评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,140评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,895评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,535评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,435评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,952评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,081评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,210评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,896评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,552评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,089评论 0 23
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,198评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,531评论 3 375
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,209评论 2 357

推荐阅读更多精彩内容