Pytania oznaczone «numpy»

149
Niezręczna macierz do tablicy

Używam Numpy. Mam macierz z 1 kolumną i N wierszami i chcę uzyskać tablicę z N elementów. Na przykład, jeśli mam M = matrix([[1], [2], [3], [4]]), chcę dostać A = array([1,2,3,4]). Aby to osiągnąć, używam A = np.array(M.T)[0]. Czy ktoś zna bardziej elegancki sposób na uzyskanie tego samego...

148
Wykreślanie mapy cieplnej 2D za pomocą Matplotlib

Korzystając z Matplotlib, chcę wykreślić mapę cieplną 2D. Moje dane to tablica Numpy n-by-n, każda o wartości od 0 do 1. Tak więc dla elementu (i, j) tej tablicy chcę wykreślić kwadrat o współrzędnej (i, j) w moim mapa cieplna, której kolor jest proporcjonalny do wartości elementu w...

142
Odrętwiały, gdzie działają wiele warunków

Mam tablicę odległości zwanych odległościami. Chcę wybrać odległości, które znajdują się między dwiema wartościami. Napisałem następujący wiersz kodu, aby to zrobić: dists[(np.where(dists >= r)) and (np.where(dists <= r + dr))] Jednak wybiera to tylko dla warunku (np.where(dists...

141
Od macierzy ND do 1D

Powiedz, że mam tablicę a: a = np.array([[1,2,3], [4,5,6]]) array([[1, 2, 3], [4, 5, 6]]) Chciałbym przekonwertować go na tablicę 1D (tj. Wektor kolumnowy): b = np.reshape(a, (1,np.product(a.shape))) ale to wraca array([[1, 2, 3, 4, 5, 6]]) co nie jest tym samym, co: array([1, 2, 3, 4, 5,...

135
Iterowanie po tablicy numpy

Czy istnieje mniej szczegółowa alternatywa dla tego: for x in xrange(array.shape[0]): for y in xrange(array.shape[1]): do_stuff(x, y) Wymyśliłem to: for x, y in itertools.product(map(xrange, array.shape)): do_stuff(x, y) Co oszczędza jedno wcięcie, ale nadal jest dość brzydkie. Mam...