Convertir un tableau Python Numpy en un tableau de tableaux simples

Question

J'ai l'impression que c'est très facile, mais je n'arrive pas à trouver comment le faire. Disons que j'ai un tableau Numpy

[1,2,3,4]

Comment puis-je convertir cela en

[[1],[2],[3],[4]]

D'une manière simple ?

Gracias

Answer 1

Vous pouvez utiliser np.newaxis :

>>> a = np.array([1,2,3,4] 
array([1, 2, 3, 4])
>>> a[:,np.newaxis]
array([[1],
       [2],
       [3],
       [4]])

Answer 2

Vous pouvez utiliser numpy.reshape :

>>> import numpy as np
>>> a = np.array([1,2,3,4])
>>> np.reshape(a, (-1, 1))
array([[1],
       [2],
       [3],
       [4]])

Si vous voulez une liste python normale, utilisez list comprehension :

>>> a = np.array([1,2,3,4]) 
>>> [[x] for x in a]
[[1], [2], [3], [4]]

Answer 3

La façon la plus évidente qui vient à l'esprit est :

>>> new = []
>>> for m in a:
        new.append([m])

mais cela crée la liste normale de listes de Python, je ne suis pas sûr que ce soit ce que vous voulez...

Answer 4

>>> A = [1,2,3,4]

>>> B = [[x] for x in A]

>>> print B
[[1], [2], [3], [4]]