目录
一、切片方法
我们除了能够一次访问一个元素之外,NumPy 还提供了访问 ndarray 子集的方式,称之为切片。切片方式是在方括号里用冒号 :
分隔起始和结束索引。通常,你将遇到三种类型的切片:
1. ndarray[start:end]
2. ndarray[start:]
3. ndarray[:end]
第一种方法用于选择在start和end索引之间的元素。第二种方法用于选择从start索引开始直到最后一个索引的所有元素。第三种方法用于选择从第一个索引开始直到 end 索引的所有元素。请注意,在第一种方法和第三种方法中,结束索引不包括在内。此外注意,因为 ndarray可以是多维数组,在进行切片时,通常需要为数组的每个维度指定一个切片。
import numpy as np
# 创建一个5行5列的二维数组
arr = np.arange(25).reshape(5, 5)
print('arr = {}\n'.format(arr))
# 选取arr从第2行(含)到第5行(不含),第1列(含)到第4列(不含) 组成的ndarray
A = arr[1:4, 0:3]
print('A = {}\n'.format(A))
# 选取arr从第2行(含)到最后一行(含),第1列(含)到第4列(不含) 组成的ndarray
B = arr[1:, 0:3]
print('B = {}\n'.format(B))
#选取arr从第1行(含)到4行(不含),第3列(含)到第4列(不含) 组成的ndarray