python中数组切片

对于一维数组

L = ['Michael', 'Sarah', 'Tracy', 'Bob', 'Jack']

>>> L[:3]  # L[0:3]
['Michael', 'Sarah', 'Tracy']

>>> L[-2:] 
['Bob', 'Jack']

>>> L[-2:-1]
['Bob']

>>> L[:5:2]  # 每两个取一个
['Michael', 'Tracy', 'Jack']

---

高维数组

类似于一维中的操作，只不过要对每一维以此操作。每一维的操作用逗号隔开。

import numpy as np

arr = np.arange(12).reshape((3, 4))
print(arr)
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

# 取第2行（取1:2行，在这些行取所有列）
line2 = arr[1:2, :]
print(line2)
[[4 5 6 7]]

# 取第2列（取所有行，在这些行取1:2列）
col2 = arr[:, 1:2]
print(col2)
[[1]
 [5]
 [9]]

# 取1:3行中2:4列的数据
data = arr[1:3, 2:4]
print(data)
[[ 6  7]
 [10 11]]