Выбор части массива Numpy - PullRequest
       9

Выбор части массива Numpy

0 голосов
/ 02 ноября 2018

Я новичок в Python, и я прочитал код, написанный кем-то опытным. Эта часть должна принимать участие в массиве Numpy

a=np.random.random((10000,32,32,3)) # random values as an example
mask=list(range(5000))
a=a[mask]

Для меня это выглядит довольно расточительно - создать еще один список, чтобы получить часть массива. Более того, результирующий массив - это действительно первые 5000 полей, сложный выбор не требуется.

Насколько я знаю, следующий код должен давать тот же результат:

a=a[:5000]

В чем преимущество первого примера? Это быстрее? Или я что-то пропустил?

...