NumPy 数组基础

import numpy as np

版本

np.__version__

'1.16.2'

整形数组

np.array([1, 2, 3, 4, 5], dtype='int8')

array([1, 2, 3, 4, 5], dtype=int8)

浮点型数组

np.array([1, 2, 3, 4, 5], dtype='float32')

array([1., 2., 3., 4., 5.], dtype=float32)

嵌套列表构成的多维数组，内层的列表被当作二维数组的行

np.array([range(i, i+3) for i in [2, 4, 6]])

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

创建一个长度为10的数组，数组的值都是0

np.zeros(10, dtype='int8')

array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0], dtype=int8)

np.full(10, 0, dtype='int8')

array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0], dtype=int8)

创建一个长度为10的数组，数组的值都是1

np.ones(10, dtype='float32')

array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32)

np.full(10, 1, dtype='float32')

array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], dtype=float32)

创建一个3×5的浮点型数组，数组的值都是3.14

np.full((3, 5), 3.14, dtype='float32')

array([[3.14, 3.14, 3.14, 3.14, 3.14],
       [3.14, 3.14, 3.14, 3.14, 3.14],
       [3.14, 3.14, 3.14, 3.14, 3.14]], dtype=float32)

创建一个数组，数组的值是一个线性序列

# 从0开始，到20结束，步长为2
# （它和内置的range()函数类似）
np.arange(0, 20, 2)

array([ 0,  2,  4,  6,  8, 10, 12, 14, 16, 18])

创建一个5个元素的数组，这5个数均匀地分配到0~1

np.linspace(0, 1, 5)

array([0.  , 0.25, 0.5 , 0.75, 1.  ])

创建一个3×3的、在0~1均匀分布的随机数数组