如何使用ctypes将Python列表列表转换为C数组?

时间:2013-10-17 18:05:39

标签: python c list multidimensional-array ctypes

如此处How do I convert a Python list into a C array by using ctypes?所示,此代码将采用python数组并将其转换为C数组。

import ctypes
arr = (ctypes.c_int * len(pyarr))(*pyarr)

对列表列表或列表列表执行相同操作的方法是什么?

例如,对于以下变量

list3d = [[[40.0, 1.2, 6.0, 0.3], [50.0, 4.2, 0, 0]], [[40.0, 1.2, 6.0, 0.3], [50.0, 4.2, 0, 0]], [[40.0, 1.2, 6.0, 0.3], [50.0, 4.2, 0, 0]]]

我试过以下但没有运气:

([[ctypes.c_double * 4] *2]*3)(*list3d)
# *** TypeError: 'list' object is not callable

(ctypes.c_double * 4 *2 *3)(*list3d)
# *** TypeError: expected c_double_Array_4_Array_2 instance, got list

谢谢!

编辑:只是为了澄清,我试图得到一个包含整个多维数组的对象,而不是一个对象列表。该对象的引用将是对期望3D数组的C DLL的输入。

2 个答案:

答案 0 :(得分:2)

works with tuples如果你不介意先做一点转换:

from ctypes import *

list3d = [
    [[0.0, 1.0, 2.0, 3.0], [4.0, 5.0, 6.0, 7.0]], 
    [[0.2, 1.2, 2.2, 3.2], [4.2, 5.2, 6.2, 7.2]],
    [[0.4, 1.4, 2.4, 3.4], [4.4, 5.4, 6.4, 7.4]],
]

arr = (c_double * 4 * 2 * 3)(*(tuple(tuple(j) for j in i) for i in list3d))

检查它是否按行主顺序正确初始化:

>>> (c_double * 24).from_buffer(arr)[:]
[0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 
 0.2, 1.2, 2.2, 3.2, 4.2, 5.2, 6.2, 7.2, 
 0.4, 1.4, 2.4, 3.4, 4.4, 5.4, 6.4, 7.4]

或者您可以创建一个空数组并使用循环初始化它。在enumerate的行和列上list,并将数据分配给切片:

arr = (c_double * 4 * 2 * 3)()

for i, row in enumerate(list3d):
    for j, col in enumerate(row):
        arr[i][j][:] = col

答案 1 :(得分:0)

我做了相应的更改

a = [[[40.0, 1.2, 6.0, 0.3], [50.0, 4.2, 0, 0]], [[40.0, 1.2, 6.0, 0.3], [50.0, 4.2, 0, 0]], [[40.0, 1.2, 6.0, 0.3], [50.0, 4.2, 0, 0]]]
arr = (((ctypes.c_float * len(a[0][0])) * len(a[0])) * len(a))
arr_instance=arr()
for i in range(0,len(a)):
  for j in range(0,len(a[0])):
    for k in range(0,len(a[0][0])):
      arr_instance[i][j][k]=a[i][j][k]

arr_instance就是你想要的。