当前位置:首页 » 《随便一记》 » 正文

Python中数组切片的用法详解

7 人参与  2023年04月01日 17:22  分类 : 《随便一记》  评论

点击全文阅读


Python中数组切片的用法详解

一、python中“::-1”代表什么?二、python中“:”的用法三、python中数组切片三、numpy中的整数数组索引四、numpy中借助【切片 : 或 …与索引数组】组合进行复杂切片五、布尔索引六、花式索引

一、python中“::-1”代表什么?

在Python中::-1表示将字符或数字倒序输出(也称【反转】)。

举个栗子,当line = "abcde"时,使用语句line[::-1],最后的运行结果为:‘edcba’。

二、python中“:”的用法

在Python中a[i:j]表示复制字符串或数字从a[i]到a[j-1](也称【切片】)。 当切片中,ij 的位置被“:”替换时,切片结果如下:

i缺省时,默认为i=0,即 a[:3]相当于 a[0:3];当j缺省时,默认为j=len(a), 即a[1:]相当于a[1:10];当i,j都缺省时,a[::]就相当于完整复制一份a。

备注:上例中,假设 a = [0,1,2,3,4,5,6,7,8,9]。

参考链接:python中::-1代表什么?

三、python中数组切片

1、NumPy 数组正切片的规则:

python 中【切片】的意思是将元素从一个给定的索引带到另一个给定的索引。

我们像这样传递切片而不是索引:[start:end]我们还可以定义步长,如下所示:[start:end:step]

备注:

如果我们不传递 start,则将其视为 0。如果我们不传递 end,则视为该维度内数组的长度。如果我们不传递 step,则视为 1。

numpy中一维数组切片:

a=np.array([1,2,3,4,5])print(a)>>>array([1, 2, 3, 4, 5])# 切片print(a[0])#查询>>>1print(a[1:3])#切片>>>[2 3]

numpy中二维数组切片:

a = np.array([    [1,2,3,4],    [5,6,7,8],    [9,10,11,12]])print(a[0,3])#第一行,第四列>>>4print(a[:,3])#第四列>>>[ 4  8 12]print(a[0,:])#第一行>>>[1 2 3 4]"""对数组使用均值函数mean()"""print(a.mean(axis=1))#计算同一列下,每一行各数字的平均值>>>[ 2.5  6.5 10.5]print(a.mean(axis=0))#计算同一行下,每一列各数字的平均值>>>[5. 6. 7. 8.]import numpy as nparr = np.array([1, 2, 3, 4, 5, 6, 7])print(arr[1:5]) # 裁切索引 1 到索引 5(不包括)的元素>>>[2 3 4 5]print(arr[4:]) # 裁切数组中索引 4 到结尾的>>>[5 6 7]print(arr[:4]) # 裁切从开头到索引 4(不包括)的元素>>>[1 2 3 4]

2、NumPy 数组的负切片的规则:

使用减号运算符从末尾开始引用索引:[-start:-end]
import numpy as nparr = np.array([1, 2, 3, 4, 5, 6, 7])# 从末尾开始的索引 3 到末尾开始的索引 1,对数组进行切片:print(arr[-3:-1])>>>[5 6]

3、NumPy 数组的使用【STEP步长】切片的规则:

使用 step 值确定切片的步长:[start: end: step]
import numpy as nparr = np.array([1, 2, 3, 4, 5, 6, 7])print(arr[1:5:2]) # 从索引 1 到索引 5,返回相隔的元素>>>[2 4]print(arr[::2])# 返回数组中相隔的元素>>>[1 3 5 7]

3、NumPy 数组中 2-D 数组的切片规则:

从第二个元素开始,对从索引 1 到索引 4(不包括)的元素进行切片。 结果的示例如下:
import numpy as nparr = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])print(arr[1, 1:4]) # 从第二个元素开始,对从索引 1 到索引 4(不包括)的元素进行切片>>>[7 8 9]

NumPy 比一般的 Python 序列提供更多的索引方式。除了之前看到的用整数和切片的索引外,数组可以由整数数组索引布尔索引花式索引

三、numpy中的整数数组索引

numpy中的整数数组索引的切片规则:

以下实例获取数组中(0,0),(1,1)和(2,0)位置处的元素。
在这里插入图片描述

import numpy as npx = np.array([[1,2],[3,4],[5,5]])y = x[[0,1,2],[0,1,0]]print(y)>>>[1 4 5]

在这里插入图片描述

x = np.array([[0,1,2],[3,4,5],[6,7,8],[9,10,11]])x>>>array([[ 0,  1,  2],       [ 3,  4,  5],       [ 6,  7,  8],       [ 9, 10, 11]])# 切片模式一:输出结果写入单列表rows = np.array([0,3,0,3])cols = np.array([0,0,2,2])y = x[rows,cols]print(y)>>>[ 0  9  2 11]# 切片模式二:输出结果写入二维数组rows = np.array([[0,0],[3,3]])cols = np.array([[0,2],[0,2]])y = x[rows,cols]print(y)>>>[[ 0  2] [ 9 11]]# 切片模式二:输出结果写入2*3的数组rows = np.array([[0,0,1],[3,2,3]])cols = np.array([[0,2,1],[0,1,2]])y = x[rows,cols]print(y)>>>array([[0, 2, 1],       [0, 1, 2]])

四、numpy中借助【切片 : 或 …与索引数组】组合进行复杂切片

借助切片 : 或 … 与索引数组组合。如下面实例:

a = np.array([[1,2,3],[4,5,6],[7,8,9]])a>>>array([[1, 2, 3],       [4, 5, 6],       [7, 8, 9]])b = a[1:3,1:3]b>>>array([[5, 6],       [8, 9]])c = a[1:3,[1,2]]c>>>array([[5, 6],       [8, 9]])d = a[...,1:]   #  arr[..., 1] 等价于 arr[:, :, 1]d>>>array([[2, 3],       [5, 6],       [8, 9]])

五、布尔索引

我们可以通过一个布尔数组来索引目标数组。
布尔索引通过布尔运算(如:比较运算符)来获取符合指定条件的元素的数组

一、以下实例获取大于 5 的元素:import numpy as npx = np.array([[ 0, 1, 2],[ 3, 4, 5],[ 6, 7, 8],[ 9, 10, 11]])print (x)>>> [[ 0 1 2] [ 3 4 5] [ 6 7 8] [ 9 10 11]]print (x[x > 5]) # 现在我们会打印出大于 5 的元素>>>[ 6 7 8 9 10 11]二、以下实例使用了 ~(取补运算符)来过滤 NaN。import numpy as npa = np.array([np.nan, 1,2,np.nan,3,4,5]) print (a[~np.isnan(a)])>>> [ 1. 2. 3. 4. 5.]三、以下实例演示如何从数组中过滤掉非复数元素。import numpy as npa = np.array([1, 2+6j, 5, 3.5+5j]) print (a[np.iscomplex(a)])>>> [2.0+6.j 3.5+5.j]

六、花式索引

花式索引指的是利用整数数组进行索引。

花式索引根据索引数组的值作为目标数组的某个轴的下标来取值。

对于使用一维整型数组作为索引,如果目标是一维数组,那么索引的结果就是对应位置的元素;如果目标是二维数组,那么就是对应下标的行。
import numpy as npx=np.arange(32).reshape((8,4))print (x[[4,2,1,7]]) # 传入顺序索引数组>>> [[16 17 18 19] [ 8 9 10 11] [ 4 5 6 7] [28 29 30 31]]print (x[[-4,-2,-1,-7]]) # 传入倒序索引数组>>> [[16 17 18 19] [24 25 26 27] [28 29 30 31] [ 4 5 6 7]]print (x[np.ix_([1,5,7,2],[0,3,1,2])]) # 传入多个索引数组(要使用np.ix_)>>> [[ 4 7 5 6] [20 23 21 22] [28 31 29 30] [ 8 11 9 10]]

原理:

np.ix_函数就是输入两个数组,产生笛卡尔积的映射关系

举个例子:
将输入数组[1,5,7,2]和数组[0,3,1,2]产生笛卡尔积,就是得到(1,0),(1,3),(1,1),(1,2);(5,0),(5,3),(5,1),(5,2);(7,0),(7,3),(7,1),(7,2);(2,0),(2,3),(2,1),(2,2);
就是按照坐标(1,0),(1,3),(1,1),(1,2)取得 x所对应的元素4,7,5,6,(5,0),(5,3),(5,1),(5,2)取得 x 所对应的元素20,23,21,22…以此类推。

import numpy as np x=np.arange(32).reshape((8,4))print (x[np.ix_([1,5,7,2],[0,3,1,2])])>>>[[ 4  7  5  6] [20 23 21 22] [28 31 29 30] [ 8 11  9 10]]

点击全文阅读


本文链接:http://zhangshiyu.com/post/57839.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

关于我们 | 我要投稿 | 免责申明

Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1