Ajouter tableau 2D à la 3D, tableau, l'extension de la troisième dimension

J'ai un tableau A qui a la forme (480, 640, 3), et un tableau B la forme de la (480, 640).

Comment puis-je ajouter ces deux comme une matrice de la forme de la (480, 640, 4)?

J'ai essayé np.append(A,B) mais il ne tient pas la dimension, tandis que le axis option provoque l' ValueError: all the input arrays must have same number of dimensions.

InformationsquelleAutor trminh89 | 2015-12-18
  1. 19

    Utilisation dstack:

    >>> np.dstack((A, B)).shape
(480, 640, 4)

    Cela gère le cas où les tableaux ont des nombres différents de dimensions et de meules de la tableaux le long du troisième axe.

    Sinon, utiliser append ou concatenate, vous aurez à faire B en trois dimensions vous-même et spécifier l'axe que vous souhaitez les rejoindre sur:

    >>> np.append(A, np.atleast_3d(B), axis=2).shape
(480, 640, 4)
    • Ou d'utiliser la plupart des méthodes de base: np.concatenate(A, B[...,None], axis=2)
    • Erreur TypeError: concatenate() takes at most 2 arguments (3 given)
    • avez-vous regarder le concatenate docs? Elles ont la priorité sur mes vieux commentaires. En tout cas, je l'aurais écrit np.concatenate([A, B[...,None]], axis=2). C'est, mettre les tableaux dans une liste
    • avez-vous regarder le concatenate docs? Elles ont la priorité sur mes vieux commentaires. En tout cas, je l'aurais écrit np.concatenate([A, B[...,None]], axis=2). C'est, mettre les tableaux dans une liste
    InformationsquelleAutor Alex Riley