NumPy 数组切片

数组切片

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

像这样传递切片而不是索引:[startend]

还可以定义步长,如下所示:[startendstep]

如果不传递 start,则将其视为 0。

如果不传递 end,则视为该维度内数组的长度。

如果不传递 step,则视为 1。

从下面的数组中截取索引 1 到索引 5 的元素:

  1. import numpy as np
  2. arr = np.array([1, 2, 3, 4, 5, 6, 7])
  3. print(arr[1:5])

注释:结果包括了开始索引,但不包括结束索引。

截取数组中索引 4 到结尾的元素:

  1. import numpy as np
  2. arr = np.array([1, 2, 3, 4, 5, 6, 7])
  3. print(arr[4:])

截取从开头到索引 4(不包括)的元素:

  1. import numpy as np
  2. arr = np.array([1, 2, 3, 4, 5, 6, 7])
  3. print(arr[:4])

负裁切

使用减号运算符从末尾开始引用索引:

从末尾开始的索引 3 到末尾开始的索引 1,对数组进行截取:

  1. import numpy as np
  2. arr = np.array([1, 2, 3, 4, 5, 6, 7])
  3. print(arr[-3:-1])

STEP

请使用 step 值确定截取的步长:

从索引 1 到索引 5,返回相隔的元素:

  1. import numpy as np
  2. arr = np.array([1, 2, 3, 4, 5, 6, 7])
  3. print(arr[1:5:2])

返回数组中相隔的元素:

  1. import numpy as np
  2. arr = np.array([1, 2, 3, 4, 5, 6, 7])
  3. print(arr[::2])

裁切 2-D 数组

从第二个元素开始,对从索引 1 到索引 4(不包括)的元素进行切片:

  1. import numpy as np
  2. arr = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
  3. print(arr[1, 1:4])

注释:请记得第二个元素的索引为 1。

从两个元素中返回索引 2:

  1. import numpy as np
  2. arr = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
  3. print(arr[0:2, 2])

从两个元素截取索引 1 到索引 4(不包括),这将返回一个 2-D 数组:

  1. import numpy as np
  2. arr = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
  3. print(arr[0:2, 1:4])