Как сохранить список как массив NumPy в Python? - PullRequest
108 голосов
/ 10 мая 2011

Мне нужно знать, возможно ли сохранить список питонов в виде массива numPy.

Ответы [ 7 ]

138 голосов
/ 10 мая 2011

Если вы посмотрите здесь, он может сказать вам, что вам нужно знать.

http://www.scipy.org/Tentative_NumPy_Tutorial#head-d3f8e5fe9b903f3c3b2a5c0dfceb60d71602cf93

По сути, вы можете создать массив из последовательности.

from numpy import array
a = array( [2,3,4] )

Или из последовательности последовательностей.

from numpy import array
a = array( [[2,3,4], [3,4,5]] )
34 голосов
/ 10 мая 2011

ты имеешь в виду что-то подобное?

from numpy  import array
a = array( your_list )
16 голосов
/ 10 мая 2011

Вы хотите сохранить его как файл?

import numpy as np

myList = [1, 2, 3]

np.array(myList).dump(open('array.npy', 'wb'))

... и затем прочитать:

myArray = np.load(open('array.npy', 'rb'))
16 голосов
/ 10 мая 2011

Да, это:

a = numpy.array([1,2,3])
7 голосов
/ 24 февраля 2017

Вы можете использовать numpy.asarray , например, для преобразования списка в массив:

>>> a = [1, 2]
>>> np.asarray(a)
array([1, 2])
4 голосов
/ 14 ноября 2014

Полагаю, вы имеете в виду преобразование списка в массив numpy? Тогда,

import numpy as np

# b is some list, then ...    
a = np.array(b).reshape(lengthDim0, lengthDim1);

дает вам массив массива b в форме, заданной в reshape.

0 голосов
/ 13 сентября 2017

Вот более полный пример:

import csv
import numpy as np

with open('filename','rb') as csvfile:
     cdl = list( csv.reader(csvfile,delimiter='\t'))
     print "Number of records = " + str(len(cdl))

#then later

npcdl = np.array(cdl)

Надеюсь, это поможет !!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...