NumPy 数组切片

---

来源：Python Numpy 教程

章节

Numpy 介绍
Numpy 安装
NumPy ndarray
NumPy 数据类型
NumPy 数组创建
NumPy 基于已有数据创建数组
NumPy 基于数值区间创建数组
NumPy 数组切片
NumPy 广播
NumPy 数组迭代
NumPy 位运算
NumPy 字符串函数
NumPy 数学函数
NumPy 统计函数
NumPy 排序、查找、计数
NumPy 副本和视图
NumPy 矩阵库函数
NumPy 线性代数

---

要访问或修改ndarray数组中的元素，可以使用数组切片/索引。

如前所述，ndarray数组索引从0开始。

使用切片访问数组，首先通过内置slice函数创建一个切片对象，该对象存储了创建时传入的start、stop和step参数，把这个切片对象传给数组，我们就可以提取数组的一部分(切片)返回。

一维数组

示例

import numpy as np 
a = np.arange(10) 
s = slice(2,7,2) 
print(a[s])

输出

[2  4  6]

上面的例子中，首先使用arange()函数创建ndarray数组，然后定义一个切片对象，它的start、stop和step值分别是2、7和2。当把这个切片对象传递给ndarray数组时，会从数组中提取一部分返回，返回部分的索引从2到7，步长2。

可以直接向ndarray数组传入start:stop:step格式的切片参数，效果相同。

示例

import numpy as np 
a = np.arange(10) 
b = a[2:7:2] 
print(b)

输出

[2  4  6]

• [i] 如果只放一个参数，则返回与索引对应的单个项。
• [:i] 如果在前面插入:，则从该索引之前的所有项都将被提取。
• [start:stop] 如果使用了两个参数，中间用:分隔，则使用默认步骤1，提取两个索引(不包括stop)之间的项。

示例

单项切片

import numpy as np 
a = np.arange(10) 
b = a[5] 
print(b)

输出

5

示例

从索引之后切片

import numpy as np 
a = np.arange(10) 
print(a[2:])

输出

[2  3  4  5  6  7  8  9]

示例

从索引之前切片

import numpy as np 
a = np.arange(10) 
print(a[:2])

输出

[0 1]

示例

在索引之间切片

import numpy as np 
a = np.arange(10) 
print(a[2:5])

输出

[2  3  4] 

上述规则也适用于多维数组。

多维数组

示例

import numpy as np 
a = np.array([[1,2,3],[3,4,5],[4,5,6]]) 
print(a)  

# 切片从索引开始
print('a[1:]切片 ') 
print(a[1:])

输出

[[1 2 3]
 [3 4 5]
 [4 5 6]]

a[1:]切片
[[3 4 5]
 [4 5 6]]

切片还可以包括省略号(…)，表示在该维度上选择全部。

示例

# 演示数组
import numpy as np 
a = np.array([[1,2,3],[3,4,5],[4,5,6]]) 

print ('数组:') 
print (a) 
print ('\n')  

# 返回一个数组，由第二列中的项目组成
print ('第二列中的项目是:')  
print (a[...,1]) 
print ('\n')  

# 返回第二行
print ('第二行中的项目是:') 
print (a[1,...]) 
print ('\n')  

# 返回第一列之后的项目
print ('第一列之后的项目:') 
print (a[...,1:])

输出

数组:
[[1 2 3]
 [3 4 5]
 [4 5 6]] 
 
第二列中的项目是:
[2 4 5] 

第二行中的项目是:
[3 4 5]

第一列之后的项目:
[[2 3]
 [4 5]
 [5 6]]