从Numpy 数组通过索引获取元素或切片

a=np.arange(20).reshape([4,5])
>>> a
array([[ 0,  1,  2,  3,  4],

       [ 5,  6,  7,  8,  9],

       [10, 11, 12, 13, 14],

       [15, 16, 17, 18, 19]])
>>> a[2,3]    #取一个标量元素
13
>>> a[[2]]    #取第三行,得到一个二维numpy数组,只有一个元素(一维数组)

array([[10, 11, 12, 13, 14]])
>>> a[:,[2]]   #取第三列

array([[ 2],

       [ 7],

       [12],

       [17]])
>>> a[1:3,[2,3]]   #取连续几行的选定列

array([[ 7,  8],

       [12, 13]])
>>> a[[1,3],[2,3]]         #要同时取选定行,选定列结果会和理解的不一样

array([ 7, 18])
>>> b=a[[1,3]]          #正确的方法是先取行,在取列
>>> b

array([[ 5,  6,  7,  8,  9],

       [15, 16, 17, 18, 19]])

>>> b[:,[2,3]]

array([[ 7,  8],

       [17, 18]]

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容