NumPy基础之花式索引

1 NumPy基础之花式索引

NumPy的花式索引(Fancy indexing)指ndarray数组使用整数数组进行索引。这的整数数组可以是python的列表等可迭代对象,也可以是NumPy数组。

花式索引,用整数数组的元素作为对应轴的索引,并且按数组元素顺序选取子集。

1.1 一个花式索引

一个花式索引,则整数数组元素为0轴的索引。

python 复制代码
>>> import numpy as np
>>> ar1=np.empty((8,5))
>>> for i in range(8):ar1[i]=i+1
>>> ar1
array([[1., 1., 1., 1., 1.],
       [2., 2., 2., 2., 2.],
       [3., 3., 3., 3., 3.],
       [4., 4., 4., 4., 4.],
       [5., 5., 5., 5., 5.],
       [6., 6., 6., 6., 6.],
       [7., 7., 7., 7., 7.],
       [8., 8., 8., 8., 8.]])
# 花式索引,使用整数数组作为轴的索引,并且按元素先后顺序返回对应轴数据
# 使用列表作为花式索引
# 返回0轴索引为5,2,0,3的元素,并且按元素先后顺序返回
>>> ar1[[5,2,0,3]]
array([[6., 6., 6., 6., 6.],
       [3., 3., 3., 3., 3.],
       [1., 1., 1., 1., 1.],
       [4., 4., 4., 4., 4.]])
# 负数作为索引,从最后一个开始向前索引为-1,-2,-3,依次类推
>>> ar1[[-3,-6,-1]]
array([[6., 6., 6., 6., 6.],
       [3., 3., 3., 3., 3.],
       [8., 8., 8., 8., 8.]])
# 使用array作为花式索引
>>> ar1[np.array((2,5,0))]
array([[3., 3., 3., 3., 3.],
       [6., 6., 6., 6., 6.],
       [1., 1., 1., 1., 1.]])

1.2 多个花式索引

多个花式索引,每个花式索引依次对应0轴、1轴...的索引,最终取0轴和1轴的交集部分。

python 复制代码
>>> import numpy as np
# ndarray.reshape()将ndarray转换为特定形状的ndarray
>>> ar2=np.arange(1,41).reshape((8,5))
>>> ar2
array([[ 1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10],
       [11, 12, 13, 14, 15],
       [16, 17, 18, 19, 20],
       [21, 22, 23, 24, 25],
       [26, 27, 28, 29, 30],
       [31, 32, 33, 34, 35],
       [36, 37, 38, 39, 40]])
# 多个花式索引,每个花式索引依次对应0轴、1轴...的索引
# [1,5,6,3]为0轴索引,[2,1,0,3]为1轴索引
# 最终取0轴和1轴的交集部分
>>> ar2[[1,5,6,3],[2,1,0,3]]
array([ 8, 27, 31, 19])
相关推荐
鸡鸭扣1 小时前
Docker:3、在VSCode上安装并运行python程序或JavaScript程序
运维·vscode·python·docker·容器·js
paterWang2 小时前
基于 Python 和 OpenCV 的酒店客房入侵检测系统设计与实现
开发语言·python·opencv
东方佑2 小时前
使用Python和OpenCV实现图像像素压缩与解压
开发语言·python·opencv
神秘_博士3 小时前
自制AirTag,支持安卓/鸿蒙/PC/Home Assistant,无需拥有iPhone
arm开发·python·物联网·flutter·docker·gitee
Moutai码农4 小时前
机器学习-生命周期
人工智能·python·机器学习·数据挖掘
小白教程5 小时前
python学习笔记,python处理 Excel、Word、PPT 以及邮件自动化办公
python·python学习·python安装
武陵悭臾5 小时前
网络爬虫学习:借助DeepSeek完善爬虫软件,实现模拟鼠标右键点击,将链接另存为本地文件
python·selenium·网络爬虫·pyautogui·deepseek·鼠标右键模拟·保存链接为htm
代码猪猪傻瓜coding5 小时前
关于 形状信息提取的说明
人工智能·python·深度学习
码界筑梦坊7 小时前
基于Flask的第七次人口普查数据分析系统的设计与实现
后端·python·信息可视化·flask·毕业设计
微笑的Java7 小时前
Python - 爬虫利器 - BeautifulSoup4常用 API
开发语言·爬虫·python