2015-12-18 1 views
3

J'ai un ensemble de numpy.arrays de NXM (deux dimensions: Range et Azimuth). J'ai besoin de former une pile de trois dimensions et d'extraire un vecteur de dimension unique pour calculer une matrice de covariance (les vecteurs rouges dans l'image).Comment extraire un vecteur du tableau numpy 3d?

enter image description here

Comment je fais efficace et facile en Python?

Répondre

3

Vous pouvez faire un tableau de numpy 3D assez facilement et il suffit d'utiliser l'indexation pour extraire les bits qui vous intéressent:

stackOfImages = np.array((image1, image2)) #iterate over these if many more 
redData = stackOfImages[:, N-1, M-1] 
+0

A ce moment, j'ai des tableaux 2D de numpy. J'ai besoin de former un tableau numpy 3D pour faire ce que vous dites? Avec hstack? – eduardosufan

+0

Désolé, j'ai complètement mal lu, j'espère que cela répond à votre question maintenant. – ncfirth

+0

Comment je peux itérer sur "k" tableaux numpy 2D pour créer un stackOfImages paramétré efficacement un pythonicly ?: stackOfImages = np.array ((image1, image2)) #iterate sur ceux-ci si beaucoup plus – eduardosufan