NumPy模块(学习笔记3-选取数组元素)

import numpy as  np

#一维数组单个元素的选取,直接根据索引值选取
a = np.array([1,2,3,4,5,6])
print(a[1],a[2])            #2 3
print(a[-1],a[-2])         #6 5
#一维数组连续元素的选取，切片，左闭右开
print(a[0:3])                #[1 2 3]
print(a[:])                   #[1 2 3 4 5 6]
print(a[0:])                 #[1 2 3 4 5 6]
print(a[-3:-1])            #[4 5]
print(a[:-1])               #[1 2 3 4 5]
#一维数组不连续元素的选取,加入步长,默认为1
print(a[0:5:2])           #[1 3 5]
print(a[::2])               #[1 3 5]
print(a[2::])               #[3 4 5 6]
print(a[:2:])               #[1 2]
print(a[5:0:-2])          #[6 4 2]

#二维数组单个元素的选取,直接根据索引值选取
b = np.array([[1,2,3],[4,5,6],[7,8,9]])
print(b[0][0],b[1][1])              #1 5

#选取单行或者单列元素，直接指定这一行列的索引即可
print(b[0])         #[1 2 3]
print(b[0,:])       #表示选取所有列[1 2 3]
print(b[:,0])       #表示选取所有行[1 4 7]

#选取某些行，列，“切片”左闭右开
print(b[0:2])          #选取第一行和第二行
print(b[:2])            #选取第一行和第二行
print(b[:,0:2])         #选取第一列和第二列
print(b[:,0:])          #选取所有列

#同时选取行列
print(b[0:2,0:2])           #同时选取第一二行，第一二列
print(b[1:3,1:3])           #同时选取第二三行，第二三列