Python的数据切片操作

• 1 一维数组的切片操作
• • 1.1 A[i]
  • 1.2 A[-1]
  • 1.3 A[:n]
  • 1.4 A[:-1]
  • 1.5 A[n:]
  • 1.6 A[-1:]
  • 1.7 A[m,n]
• 2 二维数组的切片操作
• • 2.1 B[1,:]
  • 2.2 B[:,1]
  • 2.3 B[0,2]
  • 2.4 B[1:, 1:]

A = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] 

• 作用：取数组中下标为 i 的元素

A = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] 
A[4]
Out[3]: 4

• 作用：取数组中最后一个元素

A = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] 
A[-1]
Out[4]: 9

• 作用：取从首项至下标为 n-1 的元素

A = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] 
A[:6]
Out[5]: [0, 1, 2, 3, 4, 5]

• 作用：取数组中除最后一个元素外的所有元素

A = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] 
A[:-1]
Out[6]: [0, 1, 2, 3, 4, 5, 6, 7, 8]

• 作用：取下标为n的元素至最后一项元素

A = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] 
A[4:]
Out[7]: [4, 5, 6, 7, 8, 9]

• 作用：取数组中最后一个元素

A = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] 
A[-1:]
Out[8]: [9]

• 作用：取下标n至m-1的元素

A = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] 
A[2:7]
Out[10]: [2, 3, 4, 5, 6]

• 首先要确保二维数组是np.array类型的，或者是其他类型的，list类型的二维数组会报错。

import numpy as np
B = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
B = np.array(B)
B
Out[18]: 
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

• 作用：取第二行的所有元素

B[1,:]
Out[19]: array([4, 5, 6])

• 作用：取第二列的所有元素

B[:,1]
Out[21]: array([2, 5, 8])

• 作用：取第一行，第三列的元素

B[0,2]
Out[22]: 3

• 作用：取第二行到第三行，第二列到第三列的所有元素

B[1:,1:]
Out[23]: 
array([[5, 6],
       [8, 9]])