Numpy 笔记: 多维数组的切片(slicing)和索引(indexing)【转】
目录
- 切片(slicing)操作
- 索引(indexing) 操作
- 最简单的情况
- 获取多个元素
- 切片和索引的同异
切片(slicing)操作
Numpy 中多维数组的切片操作与 Python 中 list 的切片操作一样,同样由 start, stop, step 三个部分组成
- import numpy as np
- arr = np.arange(12)
- print 'array is:', arr
- slice_one = arr[:4]
- print 'slice begins at 0 and ends at 4 is:', slice_one
- slice_two = arr[7:10]
- print 'slice begins at 7 and ends at 10 is:', slice_two
- slice_three = arr[0:12:4]
- print 'slice begins at 0 and ends at 12 with step 4 is:', slice_three
- array is: [ 0 1 2 3 4 5 6 7 8 9 10 11]
- slice begins at 0 and ends at 4 is: [0 1 2 3]
- slice begins at 7 and ends at 10 is: [7 8 9]
- slice begins at 0 and ends at 12 with step 4 is: [0 4 8]
上述例子是一维数组的例子,如果是多维数组,将不同维度上的切片操作用 逗号 分开就好了
- # coding: utf-8
- import numpy as np
- arr = np.arange(12).reshape((3, 4))
- print 'array is:'
- print arr
- # 取第一维的索引 1 到索引 2 之间的元素,也就是第二行
- # 取第二维的索引 1 到索引 3 之间的元素,也就是第二列和第三列
- slice_one = arr[1:2, 1:3]
- print 'first slice is:'
- print slice_one
- # 取第一维的全部
- # 按步长为 2 取第二维的索引 0 到末尾 之间的元素,也就是第一列和第三列
- slice_two = arr[:, ::2]
- print 'second slice is:'
- print slice_two
- array is:
- [[ 0 1 2 3]
- [ 4 5 6 7]
- [ 8 9 10 11]]
- first slice is:
- [[5 6]]
- second slice is:
- [[ 0 2]
- [ 4 6]
- [ 8 10]]
对于 slice_two,如果 arr 是用 Python 的 list 表示的,那么要得到相同的结果得像下面这样,相对来说就麻烦多了:
- import numpy as np
- arr = np.arange(12).reshape((3, 4)).tolist()
- slice_two = [
- row[::2] for row in arr
- ]
- print slice_two
- [[0, 2], [4, 6], [8, 10]]
对于维数超过 3 的多维数组,还可以通过 '…' 来简化操作
- # coding: utf-8
- import numpy as np
- arr = np.arange(24).reshape((2, 3, 4))
- print arr[1, ...] # 等价于 arr[1, :, :]
- print arr[..., 1] # 等价于 arr[:, :, 1]
- [[12 13 14 15]
- [16 17 18 19]
- [20 21 22 23]]
- [[ 1 5 9]
- [13 17 21]]
索引(indexing) 操作
最简单的情况
对于一个多维数组来说,最简单的情况就是访问其中一个特定位置的元素了,如下所示:
- # coding: utf-8
- import numpy as np
- arr = np.array([
- [1, 2, 3, 4],
- [2, 4, 6, 8],
- [3, 6, 9, 12],
- [4, 8, 12, 16]
- ])
- print '第二行第二列的值:', arr[1, 1]
- 第二行第二列的值: 4
相比之下,如果用 Python 的 list 来表示上述二维数组,获取同一个位置的元素的方法为:
- # coding: utf-8
- arr = [
- [1, 2, 3, 4],
- [2, 4, 6, 8],
- [3, 6, 9, 12],
- [4, 8, 12, 16]
- ]
- print '第二行第二列的值:', arr[1][1]
- try:
- print '第二行第二列的值(尝试用 Numpy 的方式获取):', arr[1, 1]
- except Exception as e:
- print str(e)
- 第二行第二列的值: 4
- 第二行第二列的值(尝试用 Numpy 的方式获取): list indices must be integers, not tuple
如果只是二维数组,这种差别可能看起来并不大,但想象一下假如有一个 10 维的数组,用 Python 的标准做法需要写 10 对中括号,而用 Numpy 依然只需要一对。
获取多个元素
事实上,在 Numpy 的索引操作方式 `x = arr[obj]` 中, obj 不仅仅可以是一个用逗号分隔开的数字序列,还可以是更复杂的内容。
用逗号分隔的数组序列
- 序列的长度和多维数组的维数要一致
- 序列中每个数组的长度要一致
- import numpy as np
- arr = np.array([
- [1, 2, 3, 4],
- [2, 4, 6, 8],
- [3, 6, 9, 12],
- [4, 8, 12, 16]
- ])
- print arr[[0, 2], [3, 1]]
- [4 6]
以上面这个例子来说,其含义是: 选择第一行和第三行,然后对第一行选择第四列,对第三行选择第二列。
boolean/mask index
这个不太好翻译,所以就用原来的英语表达。
所谓 boolean index,就是用一个由 boolean 类型值组成的数组来选择元素的方法。比如说对下面这样多维数组
- array([[1, 2, 3, 4], [2, 4, 6, 8], [3, 6, 9, 12], [4, 8, 12, 16]])
如果要取其中 值大于 5 的元素,就可以用上 boolean index 了,如下所示:
- import numpy as np
- arr = np.array([[1, 2, 3, 4],
- [2, 4, 6, 8],
- [3, 6, 9, 12],
- [4, 8, 12, 16]])
- mask = arr > 5
- print 'boolean mask is:'
- print mask
- print arr[mask]
- boolean mask is:
- [[False False False False]
- [False False True True]
- [False True True True]
- [False True True True]]
- [ 6 8 6 9 12 8 12 16]
除了比较运算能产生 boolean mask 数组以外, Numpy 本身也提供了一些工具方法:
- numpy.iscomplex
- numpy.isreal
- numpy.isfinite
- numpy.isinf
- numpy.isnan
切片和索引的同异
切片和索引都是访问多维数组中元素的方法,这是两者的共同点,不同之处有:
- 切片得到的是原多维数组的一个 视图(view) ,修改切片中的内容会导致原多维数组的内容也发生变化
- 切片得到在多维数组中连续(或按特定步长连续)排列的值,而索引可以得到任意位置的值,自由度更大一些
不考虑第一点的话,切片的操作是可以用索引操作来实现的,不过这没有必要就是了。
对于第一点,见下面的实验:
- import numpy as np
- arr = np.arange(12).reshape(2, 6)
- print 'array is:'
- print arr
- slc = arr[:, 2:5]
- print 'slice is:'
- print slc
- slc[1, 2] = 10000
- print 'modified slice is:'
- print slc
- print 'array is now:'
- print arr
- array is:
- [[ 0 1 2 3 4 5]
- [ 6 7 8 9 10 11]]
- slice is:
- [[ 2 3 4]
- [ 8 9 10]]
- modified slice is:
- [[ 2 3 4]
- [ 8 9 10000]]
- array is now:
- [[ 0 1 2 3 4 5]
- [ 6 7 8 9 10000 11]]
转自http://www.zmonster.me/2016/03/09/numpy-slicing-and-indexing.html
Numpy 笔记: 多维数组的切片(slicing)和索引(indexing)【转】的更多相关文章
- Python数据分析 | Numpy与1维数组操作
作者:韩信子@ShowMeAI 教程地址:http://www.showmeai.tech/tutorials/33 本文地址:http://www.showmeai.tech/article-det ...
- golang多维数组的切片
通过for循环来取多维数组的切片 package main import ( "fmt" ) func main() { a := [...]string{"USA&qu ...
- 【NumPy学习指南】day4 多维数组的切片和索引
ndarray支持在多维数组上的切片操作.为了方便起见,我们可以用一个省略号(...)来 表示遍历剩下的维度. (1) 举例来说,我们先用arange函数创建一个数组并改变其维度,使之变成一个三维数组 ...
- 初识numpy的多维数组对象ndarray
PS:内容来源于<利用Python进行数据分析> 一.创建ndarray 1.array :将一个序列(嵌套序列)转换为一个数组(多维数组) In[2]: import numpy as ...
- numpy中多维数组的绝对索引
这涉及到吧多维数组映射为一维数组. 对于3维数组,有公式: def MAP(x,y,z): return y_s * z_s * x + z_s * y + z 此公式可以推广到N维 测试代码:(两个 ...
- 利用numpy实现多维数组操作图片
1.上次介绍了一点点numpy的操作,今天我们来介绍它如何用多维数组操作图片,这之前我们要了解一下色彩是由blue ,green ,red 三种颜色混合而成,0:表示黑色 ,127:灰色 ,255:白 ...
- Java 学习笔记 二维数组和对象数组
定义二维数组 int[][] a = new int[4][5]; 可以不指定列数 int[][] a = new int[4][]; 获取行 int i = a.length(); 如果使用第一个例 ...
- C语言学习笔记--多维数组和多维指针
1. 指向指针的指针 (1)指针的本质是变量,会占用一定的内存空间 (2)可以定义指针的指针来保存指针变量的地址值 (3)指针是个变量,同样也存在传值调用与传址调用 重置动态空间的大小 #includ ...
- php学习笔记-多维数组
多维数组就是有一个数组,它里面的每个元素又是一个数组. <?php $stuff =array('food'=>array('apple','orange'),'book'=>arr ...
随机推荐
- kvm 一些web管理3方工具
OpenNebula Proxmox VE
- python+unnitest时运行后不执行main函数里面的内容
1.使用工具pycharm运行unnitest程序遇到的问题 1) 问题:运行后无法生成报告:经print()发现未执行main函数里的内容 2) 原因:使用unnitest测试框架,pycharm运 ...
- centos6.8下安装破解quartus prime16.0以及modelsim ae安装
前言 装逼使用 流程 安装modelsim: 1.modelsim ae在linux下是32位的,对于64位系统需要安装32位库:yum install xulrunner.i686 2.给予权限: ...
- 【BZOJ3625】【CF438E】小朋友和二叉树 NTT 生成函数 多项式开根 多项式求逆
题目大意 考虑一个含有\(n\)个互异正整数的序列\(c_1,c_2,\ldots ,c_n\).如果一棵带点权的有根二叉树满足其所有顶点的权值都在集合\(\{c_1,c_2,\ldots ,c_n\ ...
- POI如何自动调整Excel单元格中字体的大小
问题 目的是要将Excel中的文字全部显示出来,可以设置对齐格式为[缩小字体填充],但是这样的话只能展示出一行数据,字体会变得很小.还有一种办法,设置对齐格式为[自动换行],然后让单元格中的字体自动调 ...
- sshfs && tailon
sshfs 安装yum install glib2-devel fuse-sshfs官方版本地址https://github.com/libfuse/sshfs/releases目前最新版本:wget ...
- 洛谷AT2046 Namori(思维,基环树,树形DP)
洛谷题目传送门 神仙思维题还是要写点东西才好. 树 每次操作把相邻且同色的点反色,直接这样思考会发现状态有很强的后效性,没办法考虑转移. 因为树是二分图,所以我们转化模型:在树的奇数层的所有点上都有一 ...
- 自学华为IoT物联网之路
自学华为IoT物联网之路 01 自学华为IoT物联网_01 物联网概述 02 自学华为IoT物联网_02 常见物联网通信技术 03 自学华为IoT物联网_03 公共事业物联网常见问题及解决方案 04 ...
- Mysql数据库操作笔记
如果数据库表字段存在,则删除该表 drop table if exists `table_name` 创建数据库表语句 create table `table_name`( `id` ) not n ...
- bzoj2555(后缀自动机+LCT)
题目描述 (1):在当前字符串的后面插入一个字符串 (2):询问字符串s在当前字符串中出现了几次?(作为连续子串) 你必须在线支持这些操作. 题解 做法很自然,建出后缀自动机,维护每个节点的right ...