1. 程式人生 > >在python&numpy中切片(slice)

在python&numpy中切片(slice)

對於一維陣列來說,python原生的list和numpy的array的切片操作都是相同的。無非是記住一個規則arr_name[start: end: step],就可以了。

例項:

下面是幾個特殊的例子:

  • [:]表示複製源列表
  • 負的index表示,從後往前。-1表示最後一個元素。

相對於一維陣列而言,二維(多維)陣列用的會更多。一般語法是arr_name[行操作, 列操作]
先隨機產生一個3*4的陣列。

in:arr = np.arange(12).reshape((3, 4)) 

out:
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8
, 9, 10, 11]])
  • 取行資料
arr[i, :] #取第i行資料
arr[i:j, :] #取第i行到第j行的資料
  • 取列資料(注意資料格式)
in:arr[:,0] # 取第0列的資料,以行的形式返回的
out:
array([0, 4, 8])

in:arr[:,:1] # 取第0列的資料,以列的形式返回的
out:
array([[0],
       [4],
       [8]])
  • 取一個數據塊
# 取第一維的索引1到索引2之間的元素,也就是第二行 
# 取第二維的索引1到索引3之間的元素,也就是第二列和第三列
in:arr[1:2, 1:3] 

out: 
array([[5
, 6]])  # 取第一維的全部 # 按步長為2取第二維的索引0到末尾之間的元素,也就是第一列和第三列 in: arr[:, ::2] out: array([[ 0, 2], [ 4, 6], [ 8, 10]])

轉載自:https://www.cnblogs.com/Sinte-Beuve/p/6573246.html

參考文獻