numpy: accéder à un tableau par colonne

Question

suppose que j'ai:

 test = numpy.array([[1, 2], [3, 4], [5, 6]])
 

test[i] me permet d'obtenir la ligne du tableau (p. ex. [1, 2] ). Comment puis-je accéder à cette colonne? (eg [1, 3, 5] ). serait-ce aussi une opération coûteuse?

Answer 1

 >>> test[:,0]
array([1, 3, 5])
 

De même,

 >>> test[1,:]
array([3, 4])
 

vous permet d'accéder aux lignes. Ceci est couvert dans la Section 1.4 (Indexation) de la référence NumPy. C'est rapide, du moins dans mon expérience. C'est certainement beaucoup plus rapide que d'accéder à chaque élément d'une boucle.

Answer 2

 >>> test[:,0]
array([1, 3, 5])
 

cette commande vous donne un vecteur de ligne, si vous voulez juste la passer en boucle, c'est bien, mais si vous voulez utiliser une autre matrice de dimension 3xN, vous aurez

ValueError: tous les tableaux en entrée doivent avoir le même nombre de dimensions

tandis que

 >>> test[:,[0]]
array([[1],
       [3],
       [5]])
 

vous donne un vecteur de colonne, de sorte que vous pouvez faire concaténer ou hstack opération.

par exemple

 >>> np.hstack((test, test[:,[0]]))
array([[1, 2, 1],
       [3, 4, 3],
       [5, 6, 5]])
 

Answer 3

Et si vous voulez accéder à plusieurs colonnes à la fois, vous pouvez faire:

 >>> test = np.arange(9).reshape((3,3))
>>> test
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> test[:,[0,2]]
array([[0, 2],
       [3, 5],
       [6, 8]])