Простой Python. современный стиль программирования - читать онлайн бесплатно полную версию книги . Страница 321

(4,)

Создание массива

с помощью функции arange()

Метод arange() похож на стандартный метод range(). Если вы вызовете метод

arange(), передав ему один целочисленный аргумент num, он вернет ndarray от 0

до num-1:

>>> import numpy as np

>>> a = np.arange(10)

>>> a

array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

>>> a.ndim

1

>>> a.shape

(10,)

>>> a.size

10

С помощью двух значений он создаст массив от первого элемента до последне-

го минус один:

>>> a = np.arange(7, 11)

>>> a

array([ 7, 8, 9, 10])

Вы также можете передать как третий параметр размер шага, который будет

использован вместо единицы:

>>> a = np.arange(7, 11, 2)

>>> a

array([7, 9])

До сих пор мы показывали примеры лишь с целыми числами, но метод arrange()

работает и с числами с плавающей точкой:

>>> f = np.arange(2.0, 9.8, 0.3)

>>> f

array([ 2. , 2.3, 2.6, 2.9, 3.2, 3.5, 3.8, 4.1, 4.4, 4.7, 5. ,

5.3, 5.6, 5.9, 6.2, 6.5, 6.8, 7.1, 7.4, 7.7, 8. , 8.3,

8.6, 8.9, 9.2, 9.5, 9.8])

416

Приложение В. Py в науке

И последний прием: аргумент dtype указывает функции arrange(), какого типа

значения следует создать:

>>> g = np.arange(10, 4, -1.5, dtype=np.float)

>>> g

array([ 10. , 8.5, 7. , 5.5])

Создание массива с помощью функций zeros(),

ones() и random()

Метод zeros() возвращает массив, все значения которого равны 0. В эту функцию

вам нужно передать аргумент, в котором будет указана желаемая форма массива.

Так создается одномерный массив:

>>> a = np.zeros((3,))

>>> a

array([ 0., 0., 0.])

>>> a.ndim

1