数组的索引、组合和拆分

一、数组 的索引

1、一维数组的索引

import numpy

## 创建一维数组
arr = numpy.arange(16) ## 结果 [ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15]

## 正向索引（从0开始依次0，1，2，...）
d1 = arr[4]  
print(d1)  #结果 4

## 反向索引（从-1开始依次-1，-2，-3，...）
d2 = arr[-1] 
print(d2)  #结果 15

## 多个索引用列表接收
d3 = arr[[-1,6,8]]
print(d3)  #[15  6  8]

2、一维数组切片

arr = numpy.arange(16)

## 切片，不包含结束位置
d1 = arr[2:-1]
print(d1)  #[ 2  3  4  5  6  7  8  9 10 11 12 13 14]

## 切片，不包含结束位置
d2 = arr[2:-1:3]
print(d2)  #[ 2  5  8 11 14]

3、二维数组的索引和切片

####  二维数组
arr = numpy.arange(16).reshape((4,4))
print(arr)  #[[ 0  1  2  3]
            # [ 4  5  6  7]
            # [ 8  9 10 11]
            # [12 13 14 15]]


## 获取 第一行（注：索引是从0开始的）
d1 = arr[0,:]   #第一个参数是行，第二个参数是列
print(d1