Numpy 索引

来源:互联网 发布:java设计用户管理 编辑:程序博客网 时间:2024/05/16 18:50

ndarrays可以使用x[start:stop:step]的语法.
注意在切片的时候,stop是不包含在内的

import numpy as npdata=np.array([1,2,3,4])data[0]1data[1]2data[0:1]array([1])data[0:2]array([1, 2])data[1:2]array([2])

索引子矩阵

In [1]: import numpy as np             ...: D=np.arange(1,26).reshape((5,5))   ...: s1=np.array([1,2,3])   ...: s2=np.array([1,3,4])In [2]: D[s1,s2] # s1,s2对应下标值,作为一个索引Out[2]: array([ 7, 14, 20])In [3]: D[s1][:,s2] #先取出指定行,再取出指定列Out[3]: array([[ 7,  9, 10],       [12, 14, 15],       [17, 19, 20]])
原创粉丝点击