2012-03-15 4 views
60

Я использую внешний модуль (libsvm), который не поддерживает массивы numpy, только кортежи, списки и dicts. Но мои данные находятся в массиве 2d numpy. Как я могу преобразовать его в питоновский путь, ака без петель.Преобразование массива 2d numpy в список списков

>>> 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 
+7

Вы можете проверить scikit-узнать, который включает в себя оболочку LibSVM, который делает обрабатывать Numpy массивы изначально. http://scikit-learn.org/stable/modules/classes.html#module-sklearn.svm –

ответ

92
>>> 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'> 
Смежные вопросы