从numpy数组中提取数据的子集

import numpy as np

# 创建一维数组
arr = np.array([1, 2, 3, 4, 5])

# 提取索引1到索引3的子集（不包括索引3）
subset = arr[1:3]

print(subset)  # 输出：[2 3]

import numpy as np

# 创建二维数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 提取第一行的子集
subset1 = arr[0, :]  # 或者 arr[0]
print(subset1)  # 输出：[1 2 3]

# 提取第一列的子集
subset2 = arr[:, 0]
print(subset2)  # 输出：[1 4 7]

# 提取一个子矩阵
subset3 = arr[1:3, 1:3]
print(subset3)
# 输出：
# [[5 6]
#  [8 9]]

import numpy as np

# 创建一维数组
arr = np.array([1, 2, 3, 4, 5])

# 提取索引1到索引4的子集，步长为2
subset = arr[1:5:2]

print(subset)  # 输出：[2 4]