NumPy 对数组的操作的介绍这里从
- 通过索引访问数组
- 变换数组的形态
两个方面进行
对ndarray数组的操作
通过索引访问数组
对一维数组的访问
一维数组的索引方法跟Python中的 list 的索引方法一致:
# 直接通过下标获取某个元素
arr = np.arange(10)
print(arr[5])
# 用范围作为下标获取数组的一个切片
print(arr[3:5]) # 包括 arr[3] 但不包括 arr[5]!
# 范围中的第三个参数表示步长
print(arr[1:-1:2]) # 2表示各一个元素取一个元素
对多维数组的访问
多维数组每个维度都有一个索引,各个维度的索引之间用逗号隔开
# 创建一个二维数组
arr = np.array([
[1, 2, 3, 4, 5],
[4, 5, 6, 7, 8],
[7, 8, 9, 10, 11]
])
# 获取第0行中第3列和第4列的元素
print(arr[0, 3:5]