свернуть список ndarray в матрицу - PullRequest
2 голосов
/ 13 января 2012

У меня есть список, содержащий объекты типа numpy.ndarray, все элементы списка имеют одинаковое значение .shape.

Как мне свернуть это в матрицу?

Ответы [ 2 ]

4 голосов
/ 13 января 2012

Звучит так, будто вы ищете numpy.vstack() или numpy.hstack(), в зависимости от того, хотите ли вы, чтобы массивы становились строками или столбцами матрицы.

Из руководства:

>>> a = np.array([1, 2, 3])
>>> b = np.array([2, 3, 4])
>>> np.vstack((a,b))
array([[1, 2, 3],
       [2, 3, 4]])
3 голосов
/ 13 января 2012

Я не уверен, если вы имеете в виду это:

>>> alist = [np.array([item, item+1]) for item in range(5)]
>>> alist
[array([0, 1]), array([1, 2]), array([2, 3]), array([3, 4]), array([4, 5])]

>>> np.array(alist)
array([[ 0,  1],
       [ 1,  2],
       [ 2,  3],
       [ 3,  4],
       [ 4,  5]])
>>> 

или вам нужна матрица:

>>> np.matrix(alist)
matrix([[ 0,  1],
        [ 1,  2],
        [ 2,  3],
        [ 3,  4],
        [ 4,  5]])
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...