3

据我了解，将 NumPy 数组转换为本机 Python 列表的推荐方法是使用ndarray.tolist.

唉，这在使用结构化数组时似乎无法递归工作。事实上，一些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可以预期的吗？
  • 有没有更好的方法来实现这一点？
4

1 回答 1

0

只是为了概括一下，我将在您的 dtype 中添加另一个字段

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

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

显示repr确实使用列表和元组：

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可以解决这个问题。

于 2016-09-15T04:04:49.497 回答