numpy常规操作

numpy 是Python的一个常用数值计算库。

1. numpy数组

import numpy as np
a = np.array([1, 2, 3])
# 一维数组
b = np.array([[1, 2, 3],[4, 5, 6]])
# 二维数组

2. 数组读取

b[0, 0]
# 输出结果为1
b[:,1]
# 输出结果为第2列的所有行array([2, 5])

3. 常见方法

np.shape(b)
# 输出结果为(2, 2)
np.sum(b)
# 输出结果为21
np.sum(b, axis=0)
# 输出结果为（5， 7， 9）