Déballez tableau NumPy par colonne

Si j'ai un tableau NumPy, par exemple, 5x3, est-il un moyen de décompresser la colonne par colonne, tout à la fois à transmettre à une fonction plutôt que comme ceci: my_func(arr[:, 0], arr[:, 1], arr[:, 2])?

Un peu comme *args pour la liste déballage mais par colonne.

np.hsplit(x,x.shape[1])

OriginalL'auteur jeff_new | 2014-11-20

  1. 18

    Vous pouvez décompresser la transposée de la matrice afin d'utiliser les colonnes de vos arguments de la fonction:

    my_func(*arr.T)

    Voici un exemple simple:

    >>> x = np.arange(15).reshape(5, 3)
array([[ 0,  5, 10],
       [ 1,  6, 11],
       [ 2,  7, 12],
       [ 3,  8, 13],
       [ 4,  9, 14]])

    Nous allons écrire une fonction pour ajouter les colonnes (normalement fait avec x.sum(axis=1) dans NumPy):

    def add_cols(a, b, c):
    return a+b+c

    Puis nous avons:

    >>> add_cols(*x.T)
array([15, 18, 21, 24, 27])

    Tableaux NumPy sera déballé le long de la première dimension, d'où la nécessité de transposer le tableau.

    Exactement ce dont j'avais besoin! Merci! EDIT: désolé je ne peux pas upvote encore...
    Pas de problème, content que ça a aidé!

    OriginalL'auteur Alex Riley

  2. 8

    numpy.split fend d'un tableau en plusieurs sous-ensembles. Dans votre cas, indices_or_sections est 3 car vous avez 3 colonnes, et axis = 1 puisque nous sommes fractionnement par colonne. 

    my_func(numpy.split(array, 3, 1))

    OriginalL'auteur Stephanie