Questions marquées «numpy»

150
Matrice numpy à tableau

J'utilise numpy. J'ai une matrice avec 1 colonne et N lignes et je veux obtenir un tableau à partir de N éléments. Par exemple, si je l'ai M = matrix([[1], [2], [3], [4]]), je veux obtenir A = array([1,2,3,4]). Pour y parvenir, j'utilise A = np.array(M.T)[0]. Quelqu'un connaît-il un moyen plus...

141
Des baies ND aux baies 1D

Disons que j'ai un tableau a: a = np.array([[1,2,3], [4,5,6]]) array([[1, 2, 3], [4, 5, 6]]) Je voudrais le convertir en un tableau 1D (c'est-à-dire un vecteur de colonne): b = np.reshape(a, (1,np.product(a.shape))) mais cela revient array([[1, 2, 3, 4, 5, 6]]) ce qui n'est pas la même chose que:...

139
Tracer une carte thermique 2D avec Matplotlib

En utilisant Matplotlib, je souhaite tracer une carte thermique 2D. Mes données sont un tableau Numpy n-par-n, chacun avec une valeur comprise entre 0 et 1. Donc, pour l'élément (i, j) de ce tableau, je veux tracer un carré à la coordonnée (i, j) dans mon heat map, dont la couleur est...

135
Itérer sur un tableau numpy

Existe-t-il une alternative moins verbeuse à ceci: for x in xrange(array.shape[0]): for y in xrange(array.shape[1]): do_stuff(x, y) Je suis venu avec ceci: for x, y in itertools.product(map(xrange, array.shape)): do_stuff(x, y) Ce qui économise une indentation, mais reste assez moche. J'espère...