Numpy中最常用的是ndarray数组,而因为其是多维的有时候需要在不同维度上进行切片。
其实它的切片方式与Matlab类似,熟练以后就很好掌握了。
基本原理是这样,首先切片需要使用序列生成即冒号运算符:
一个单冒号为选取整个序列,也可以指定范围,如1:5表示1~4(包括)这4个元素。同时可以指定步长如1:6:2表示选取1,3,5这3个元素,这个与Python的列表切片是一致的。
然后,使用逗号,来区分轴(axis),轴可以简单地理解为在某个维度上的投影。例如axis=0指的就是行,axis=1指的就是列。那么第一个选取就是行,第二个选取的就是列,中间用逗号隔开,例如下面的一条语句。
1 | arr[:,2] |