numpy 对二维数组的常用操作

目录
1、提取二维数组的某几列或某几行
2、获取某个范围的数据
3、所有元素求和
4、计算数组中非零元素的个数
5、使用布尔型掩码提取某些行或某些列
6、获取数组的行数或列数
7、获取最后一列（或行）的元素

1、提取二维数组的某几列或某几行
import numpy as np
# 定义 3*3 的 numpy 数组
matrix = np.array([[1, 3, 2],
[8, 0, 6],
[9, 7, 0]])
# 提取第1、3列（行的提取同理）
matrix1 = matrix[:, [0, 2]]
print(matrix1)

# ========== 结果 ========== #
[[1 2]
[8 6]
[9 0]]
2、获取某个范围的数据
# 获取第1、2行中第1、3列的数据. 注意 0:2 表示左闭右开的区间
matrix1 = matrix[0:2, [0, 2]]
print(matrix1)

# ==

numpy 对二维数组的常用操作最先出现在Python成神之路。