NumPy 结构化数组的真正递归“tolist()”

据我了解，将 NumPy 数组转换为本机 Python 列表的推荐方法是使用ndarray.tolist http://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.tolist.html.

唉，在使用结构化数组时，这似乎不能递归地工作。确实，有些ndarray对象在结果列表中被引用，未转换：

>>> dtype = numpy.dtype([('position', numpy.int32, 3)])
>>> values = [([1, 2, 3],)]
>>> a = numpy.array(values, dtype=dtype)
>>> a.tolist()
[(array([1, 2, 3], dtype=int32),)]

我确实写了一个简单的函数来解决这个问题：

def array_to_list(array):
    if isinstance(array, numpy.ndarray):
        return array_to_list(array.tolist())
    elif isinstance(array, list):
        return [array_to_list(item) for item in array]
    elif isinstance(array, tuple):
        return tuple(array_to_list(item) for item in array)
    else:
        return array

使用时，可以提供预期的结果：

>>> array_to_list(a) == values
True

这个函数的问题是它重复了ndarray.tolist通过重新创建它输出的每个列表/元组。不是最佳的。

所以问题是：

• 这种行为是ndarray.tolist值得期待吗？
• 有没有更好的方法来实现这一点？

只是为了概括一下这一点，我将在您的数据类型中添加另一个字段

In [234]: dt = numpy.dtype([('position', numpy.int32, 3),('id','U3')])

In [235]: a=np.ones((3,),dtype=dt)

The reprdisplay 确实使用列表和元组：

In [236]: a
Out[236]: 
array([([1, 1, 1], '1'), ([1, 1, 1], '1'), ([1, 1, 1], '1')], 
  dtype=[('position', '<i4', (3,)), ('id', '<U3')])

但正如你所注意到的，tolist不扩展元素。

In [237]: a.tolist()
Out[237]: [(array([1, 1, 1]), '1'), (array([1, 1, 1]), '1'), 
   (array([1, 1, 1]), '1')]

类似地，可以从完全嵌套的列表和元组创建这样的数组。

In [238]: a=np.array([([1,2,3],'str')],dtype=dt)
In [239]: a
Out[239]: 
array([([1, 2, 3], 'str')], 
  dtype=[('position', '<i4', (3,)), ('id', '<U3')])
In [240]: a.tolist()
Out[240]: [(array([1, 2, 3]), 'str')]

从这个不完整的递归中重新创建数组没有问题：

In [250]: np.array(a.tolist(),dtype=dt)
Out[250]: 
array([([1, 2, 3], 'str')], 
      dtype=[('position', '<i4', (3,)), ('id', '<U3')])

这是我第一次看到有人使用tolist有这样的结构化数组，但我并不太惊讶。我不知道开发者是否会认为这是一个错误。

为什么需要这个数组的纯列表/元组渲染？

我想知道是否有一个功能numpy/lib/recfunctions.py这解决了这个问题。