Tableaux multidimensionnels Numpy et accès inversé

Question

J'ai un tableau numpy avec rank = 4. Supposons qu'il ressemble à ceci:

np.shape(my_array) 
(10,5,25,50) 

J'ai aussi une liste de 10 valeurs, telles que

x = np.arange(10) 

Je voudrais être en mesure de tracer x contre le premier axe de my_array pour chaque valeur, qui est, je voudrais faire:

for axis2_index in xrange(5): 
    for axis3_index in xrange(25): 
     for axis4_index in xrange(50): 
      plt.plot(x,my_array[:,axis2_index,axis3_index,axis4_index]) 

Mais bien sûr, je veux le faire sous une forme numpy, sans for boucles. Cela fait longtemps que je me bats avec ça, mais je n'arrive toujours pas à trouver un moyen facile de le faire.

En d'autres termes, je voudrais pouvoir accéder au my_array pour obtenir, par exemple, une matrice de vecteurs 5x25x50 de 10 chacun, correspondant au premier axe my_array. J'ai aussi essayé np.reshape, mais cela change l'ordre des éléments de telle sorte que je ne peux pas l'utiliser.

Answer 1

Voici un exemple simple en utilisant la diffusion:

y = randn(10,4) 
x = arange(10) 

plot(x[:,None],y) 

Je suppose que dans votre cas, vous auriez x[:,None,None,None].