文章目录

  • numpy数组切片操作
  • 一维数组(冒号:)
  • 1、一个参数:a[i]
  • 2、两个参数:b=a[i:j]
  • 3、三个参数:格式b = a[i:j:s]
  • 4、例子
  • 二维数组(逗号,)
  • 取元素 X[n0,n1]
  • 切片 X[s0:e0,s1:e1]
  • 切片特殊情况 X[:e0,s1:]
  • 代码实例
  • reference


numpy数组切片操作

列表用 [ ] 标识,支持字符,数字,字符串甚至可以包含列表(即嵌套),是 python 最通用的复合数据类型。
关于索引
左到右索引默认 0 开始,从右到左索引默认 -1 开始。

python数组条件截取 numpy截取数组_python数组条件截取

一维数组(冒号:)

通过冒号分隔切片参数 start:stop:step 来进行切片操作:

import numpy as np
a=[1,2,3.4,5]
print(a)
[ 1 2 3 4 5 ]

1、一个参数:a[i]

如 [2],将返回与该索引相对应的单个元素。

2、两个参数:b=a[i:j]

b = a[i:j] 表示复制a[i]到a[j-1],以生成新的list对象
i缺省时默认为0,即 a[:n] 代表列表中的第一项到第n项,相当于 a[0:n]
j缺省时默认为len(alist),即a[m:] 代表列表中的第m+1项到最后一项,相当于a[m:5]
当i,j都缺省时,a[:]就相当于完整复制a

print(a[-1])     取最后一个元素
    结果:[5]
    
print(a[:-1])     除了最后一个取全部
    结果:[ 1 2 3 4 ]
    
print(a[1:])     取第二个到最后一个元素
    结果:[2 3 4 5]

3、三个参数:格式b = a[i:j:s]

这里的s表示步进,缺省为1.(-1时即翻转读取)
所以a[i:j:1]相当于a[i:j]

s='abcdefg'
print(s[0:4] #输出abcd
print(s[0:4:2]) #步长为 2(间隔一个位置)来截取,输出ac
当s<0时,i缺省时,默认为-1. j缺省时,默认为-len(a)-1
所以a[::-1]相当于 a[-1:-len(a)-1:-1],也就是从最后一个元素到第一个元素复制一遍。所以你看到一个倒序的东东。

print(a[::-1])     取从后向前(相反)的元素
结果:[ 5 4 3 2 1 ]
 
print(a[2::-1])     取从下标为2的元素**翻转读取**
结果:[ 3 2 1 ]

4、例子

题解:如果 l 能移动,那么 r 一定可以随便移动,如果 l 不动,那么 r 一定不能动
那么我们现在只要考虑 l 的移动即可
考虑找到位置 k 之前的最左边的最小的字符,如果存在,先手可以直接把 l 移过去,那么后手就没得走了
如果不存在,那么先手显然没得移动
所以直接根据每个位置左边最小的字符是否小于当前位置的字符即可判断胜负

m='z'
for c in input():
    print('MAinkne'[c>m::2])
    m=min(m, c)

二维数组(逗号,)

X[n0,n1]是通过 numpy 库引用二维数组或矩阵中的某一段数据集的一种写法。
类似的,X[n0,n1,n2]表示取三维数组,取N维数组则有N个参数,N-1个逗号分隔。

取元素 X[n0,n1]

这是最基本的情况,表示取 第0维 的第 n0 个元素,继续取 第1维 的第 n1个元素。如 X[2,2] 表示第0维第2个元素[20,21,22,23],然后取其第1维的第2个元素即 22;

切片 X[s0:e0,s1:e1]

这是最通用的切片操作,表示取 第0维 的第 s0 到 e0 个元素,继续取 第1维 的第 s1 到 e1 个元素(左闭右开)。如 X[1:3,1:3] 表示第0维第(1:3)个元素[[10,11,12,13],[20,21,22,23]],然后取其第1维的第(1:3)个元素即 [[11,12],[21,22]];

切片特殊情况 X[:e0,s1:]

特殊情况,即左边从0开始可以省略X[:e0,s1:e1],右边到结尾可以省略X[s0:,s1:e1],取某一维全部元素X[:,s1:e1],事实上和Python 的 序列切片规则是一样的。

常见的 X[:,0] 则表示 第0维取全部,第1维取0号元素;

代码实例

import numpy as np
X = np.array([[0,1,2,3],[10,11,12,13],[20,21,22,23],[30,31,32,33]])
#X 是一个二维数组,维度为 0 ,1;第 0 层 [] 表示第 0 维;第 1 层 [] 表示第 1 维;

# X[n0,n1] 表示第 0 维 取第n0 个元素 ,第 1 维取第 n1 个元素
print(X[1,0])
# X[1:3,1:3] 表示第 0 维 取 (1:3)元素 ,第 1 维取第(1:3) 个元素
print(X[1:3,1:3])

# X[:n0,:n1] 表示第 0 维 取 第0 到 第n0 个元素 ,第 1 维取 第0 到 第n1 个元素
print(X[:2,:2])
# X[:,:n1] 表示第 0 维 取 全部元素 ,第 1 维取 第0 到第n1 个元素
print(X[:,:2])

# X[:,0]) 表示第 0 维 取全部 元素 ,第 1 维取第 0 个元素
print(X[:,0])

输出结果:

10
[[11 12]
 [21 22]]
[[ 0  1]
 [10 11]]
[[ 0  1]
 [10 11]
 [20 21]
 [30 31]]
[ 0 10 20 30]