Numpy二维数组传递到C扩展模块

[starviolet]

在Python下,建立一个numpy二维数组, 比如 a = zero((3,5))
把a传递到C扩展模块里
在C扩展模块里面直接修改a内的数值..
扩展模块运行完毕后,Python需要重新用到修改过的a

请问怎么实现?

[notax]

#! /bin/sh

cat > mystuff.c << _EOF

void show_matrix(int *matrix, int rows, int columns)
{
    int i, j;
    for (i=0; i<rows; i++) {
        for (j=0; j<columns; j++) {
            printf("matrix[%d][%d] = %d\n", i, j, matrix[i*rows + j]);
        }
    }
}

_EOF

gcc -shared -O2 mystuff.c  -ldl -o mystuff.so


cat > test.py << _EOF

import ctypes
import numpy as np

lib = ctypes.cdll.LoadLibrary("./mystuff.so")

arr = np.zeros((3,5))
#arr = np.array([[1,2],[3,4]])
tmp = np.asarray(arr)
rows, cols = tmp.shape
dataptr = tmp.ctypes.data_as(ctypes.c_char_p)
lib.show_matrix(dataptr, rows, cols)

_EOF

python test.py

sh test.sh

[flamkuavos]

正在学习python，帮你顶了，lz100%的结贴率大家会的一定要帮帮他！

[thy38]

应该可以用SWIG，不过这个也用SWIG好像用牛刀了，我写了个包装函数，把numpy的二维数组打包成ctypes的标准数组形式，传送给C。

from ctypes import *
import numpy

def Convert1DToCArray(TYPE, ary):
    arow = TYPE(*ary.tolist())
    return arow

def Convert2DToCArray(ary):
    ROW = c_int * len(ary[0])
    rows = []
    for i in range(len(ary)):
        rows.append(Convert1DToCArray(ROW, ary[i]))
    MATRIX = ROW * len(ary)
    return MATRIX(*rows)

a=numpy.array([[1,2,2],[1,3,4]])
caa = Convert2DToCArray(a)

def ShowCArrayArray(caa):
    for row in caa:
        for col in row:
            print col,
        print

print caa
ShowCArrayArray(caa)

[thy38]

返回值的话解包就可以了，别忘了指定返回类型。

[notax]

[Quote=引用 5 楼 thy38 的回复:]

LS把二维数组转成一维数组传维C函数，思路不错，赞一个！
[/Quote]

谢谢，抛砖引玉，大概用PyArrayObject 也行吧，
想看看其他的,再比较比较,学习学习

[thy38]

LS把二维数组转成一维数组传维C函数，思路不错，赞一个！