将2d numpy 数组转换为列表列表

我使用一个外部模块(Libsvm) ，它不支持 numpy 数组，只支持 tuple、 list 和 dicts。但是我的数据是在一个二维数组中。我怎样才能转换它的蟒蛇的方式，即没有循环。

>>> import numpy
>>> array = numpy.ones((2,4))
>>> data_list = list(array)
>>> data_list
[array([ 1.,  1.,  1.,  1.]), array([ 1.,  1.,  1.,  1.])]


>>> type(data_list[0])
<type 'numpy.ndarray'>  # <= what I don't want


# non pythonic way using for loop
>>> newdata=list()
>>> for line in data_list:
...     line = list(line)
...     newdata.append(line)
>>> type(newdata[0])
<type 'list'>  # <= what I want

python
arrays
list
numpy
multidimensional-array

121980 次浏览

小开

最佳答案

You can simply cast the matrix to list with matrix.tolist(), proof:

>>> import numpy
>>> a = numpy.ones((2,4))
>>> a
array([[ 1.,  1.,  1.,  1.],
[ 1.,  1.,  1.,  1.]])
>>> a.tolist()
[[1.0, 1.0, 1.0, 1.0], [1.0, 1.0, 1.0, 1.0]]
>>> type(a.tolist())
<type 'list'>
>>> type(a.tolist()[0])
<type 'list'>