Python数据分析-NumPy模块-选取数组元素

306 阅读2分钟

一维数组的元素选取

选取单个元素

from numpy import array
a=array([1,2,5,48,62,9,4,7,2,3,6,9])
# 正序索引,其值是从0开始计数
print(a[0])
print(a[4])
# 倒序索引,其值是从-1开始计数
print(a[-1])
print(a[-5])]

结果:
在这里插入图片描述

选取连续的元素

from numpy import array
a=array([1,2,5,48,62,9,4,7,2,3,6,9])
print(a[1:6])
print(a[3:-2])
print(a[:3])
print(a[:-5])
print(a[5:])
print(a[-5:])

结果:

在这里插入图片描述

起始位置和结束位置的索引值构成的是一个“左闭右开”的区间
选取起始位置的元素,但是不选取结束位置的元素

选取不连续的元素

from numpy import array
a=array([1,2,5,48,62,9,4,7,2,3,6,9])
# 起始位置:结束位置:步长
print(a[1:5:2])
print(a[5:1:-2])
print(a[::3])
print(a[3::])
print(a[:3:])

结果:
在这里插入图片描述

二维数组的元素选取

选取单个元素

from numpy import array
a = array([[1,2,3],[4,5,6],[7,8,9],[10,11,12]])
print(a[2,2])

结果:
在这里插入图片描述

选取单行或单列的元素

from numpy import array
a = array([[1,2,3],[4,5,6],[7,8,9],[10,11,12]])
print(a[1])
print(a[:,0])

结果:
在这里插入图片描述

选取多行或多列的元素

from numpy import array
a = array([[1,2,3],[4,5,6],[7,8,9],[10,11,12]])
print(a[1:3])
print(a[:3])
print(a[1:])
# 选取多行

结果:
在这里插入图片描述

from numpy import array
a = array([[1,2,3,3],[4,5,6,3],[7,8,9,3],[10,11,12,3]])
print(a[:,1])
print(a[:,:2])
print(a[:,2:])
print(a[:,1:3])
# 选取多列

结果:
在这里插入图片描述

同时选取行列元素

from numpy import array
a = array([[1,2,3,3],[4,5,6,3],[7,8,9,3],[10,11,12,3]])
print(a[0:3,0:2])

结果:
在这里插入图片描述