一份友好的 NumPy 入门代码清单

NumPy 安装

使用 pip 可以方便地安装 NumPy。

python -m pip install numpy

NumPy 概览

# 引入 NumPy 包，习惯上引入为 np，方便书写
import numpy as np

# 将一个列表转换为 NumPy 数组
a = np.array([[1,2,3],[4,5,6]])

# 打印数组
print(a)

# 打印数组的维度、形状和大小
print('number of dim:', a.ndim)
print('shape', a.shape)
print('size:', a.size)

[[1 2 3]
 [4 5 6]]
number of dim: 2
shape (2, 3)
size: 6

NumPy 创建 Array

从 Python 列表创建

使用 np.array 从 Python 列表创建 NumPy Array

a = np.array([6,67,7])
print(a)

[6 67  7]

指定数据类型

使用 dtype 属性指定 NumPy Array 元素的类型。
例如要制定类型为整数型，则可以使用 np.int，可以使用 np.int32 进一步指定其为 32 位整数型。
从下面程序的输出结果看到，若不指定位数，np.int 默认为 64 位。

a = np.array([6,67,7], dtype=np.int)
print(a.dtype)
a = np.array([6,67,7], dtype=np.int32)
print(a.dtype)
a = np.array([6,67,7], dtype=np.int64)
print(a.dtype)

int32
int32
int64

其他类型，例如 float 同理。

a = np.array([6,67,7], dtype=np.float)
print(a.dtype)
a = np.array([6,67,7], dtype=np.float32)
print(a.dtype)
a = np.array([6,67,7], dtype=np.float64)
print(a.dtype)

float64
float32
float64

创建一个二维数组

a = np.array([[6,67,7],[6,67,7]])
print(a)

[[6 67  7]
 [6 67  7]]

全零数组

使用 np.zeros() 创建全零数组，以 (x_dim,y_dim) 的形式指定维度。

a = np.zeros((6,7) )
print(a)

[[0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0. 0. 0.]
 [0. 0. 0. 0. 0. 0. 0.]]

全一数组

使用 np.ones() 创建全一数组

a = np.ones((6,7) )
print(a