Question

我正在编写一个代码，在那里我创建了一个numpy指针数组。他们指向其他数组。

我可以成功地（没有生成异常）将指针存储在指针数组的一个元素内。但我无法将此指针重新转换为numpy数组。

当指针存储在一个numpy指针数组中时，问题就出现了。当我将指针存储在普通的pyhon变量中时，我可以成功地存储和检索数组。

请注意，由于性能原因，我不能只创建一个python指针列表。

此代码有效：

import numpy, ctypes
ic = numpy.array([[1,2],[3,4]],dtype=numpy.int32)
pointer = ic.__array_interface__['data'][0]
v = numpy.ctypeslib.as_array(ctypes.cast(pointer,ctypes.POINTER(ctypes.c_int)),shape=(2,2))
print(v)

和v将返回ic中的初始数组。

此代码不起作用：

import numpy, ctypes
ic = numpy.array([[1,2],[3,4]],dtype=numpy.int32)
pointers = numpy.zeros(5, dtype=ctypes.POINTER(ctypes.c_int))
pointers[0] = ic.__array_interface__['data'][0]
numpy.ctypeslib.as_array(ctypes.cast(pointers[0],ctypes.POINTER(ctypes.c_int)),shape=(2,2))

最后一行将提供以下异常：

File "/opt/intel/intelpython3/lib/python3.5/ctypes/__init__.py", line 484, in cast
return _cast(obj, obj, typ)
ctypes.ArgumentError: argument 1: <class 'TypeError'>: wrong type

问题：如何从numpy数组指针中存储和检索numpy-array？

Answer 1

为数组建立索引会返回一个np.int32对象，而不是本机Python int：

In [118]: type(pointer)
Out[118]: int
In [119]: type(pointers[0])
Out[119]: numpy.int32

使用item提取int：

In [120]: type(pointers[0].item())
Out[120]: int

您也可以先将数组转换为列表

In [121]: type(pointers.tolist()[0])
Out[121]: int

构建它时，

pointers是np.int32 dtype

In [123]: pointers = numpy.zeros(5, dtype=ctypes.POINTER(ctypes.c_int))
In [124]: pointers.dtype
Out[124]: dtype('int32')

或者创建一个对象dtype数组

In [125]: pointers = numpy.zeros(5, dtype=object)
In [126]: pointers[0] = pointer
In [127]: pointers
Out[127]: array([157379928, 0, 0, 0, 0], dtype=object)
In [128]: type(pointers[0])
Out[128]: int

Numpy：从指针数组中重新获取数组

此代码有效：

此代码不起作用：

1 个答案: