索引多维 numpy 数组中的多个元素
问题内容
我想使用另一个索引数组提取给定多维 numpy 数组的元素。但是它的行为并不符合我的预期。下面是一个简单的例子:
import numpy as npa = np.random.random((3, 3, 3))idx = np.asarray([[0, 0, 0], [0, 1, 2]])b = a[idx]print(b.shape) # expect (2, ), got (2, 3, 3, 3)
为什么会这样呢?我应该如何修改代码以使 b 仅包含两个元素:a[0, 0, 0] 和 a[0, 1, 2]?
正确答案
您正在寻找 numpy 高级索引
https://www.php.cn/link/2d661a763280f48803f3c9ba8ba0e00b
在您的情况下,您需要在每个轴上使用 idx:
a[idx[:,0], idx[:, 1], idx[:, 2]].shape == (2,) # True
文章推荐
-
Python实现字典的key和values的交换
有些时候我们不得已要利用values来反向查询key,有没有简单的方法呢?下面我给大家列举一些方法,方便大家使用pyth...
Python
2个月前 (02-11) 207 -
使用Python脚本来获取Cisco设备信息的示例
今天发现一个使用python写的管理cisco设备的小框架tratto,可以用来批量执行命令。下载后主要有3个文件:Sy...
Python
2个月前 (02-11) 110 -
Python的Django中django-userena组件的简单使用教程
利用twitter/bootstrap,项目的基础模板算是顺利搞定。接下来开始处理用户中心。用户中心主要包括用户登陆、注...
Python
2个月前 (02-11) 99 -
零基础写python爬虫之神器正则表达式
接下来准备用糗百做一个爬虫的小例子。但是在这之前,先详细的整理一下Python中的正则表达式的相关内容。正则表达式在Py...
Python
2个月前 (02-11) 112 -
零基础写python爬虫之抓取百度贴吧代码分享
这里就不给大家废话了,直接上代码,代码的解释都在注释里面,看不懂的也别来问我,好好学学基础知识去! 代码如下:# -*-...
Python
2个月前 (02-11) 98
最新文章
- 微信三级分销系统产品详情页怎么开发? 3周前 (03-13)
- 微信公众号-获取用户信息(网页授权获取)实现步骤 3周前 (03-13)
- php微信公众号开发(2)百度BAE搭建和数据库使用 3周前 (03-13)
- php微信公众号开发(3)php实现简单微信文本通讯 3周前 (03-13)
- php微信公众号开发(4)php实现自定义关键字回复 3周前 (03-13)
- php版微信小店调用api示例代码 3周前 (03-13)
- php版微信小店API二次开发及使用示例 3周前 (03-13)
- C# 微信公众号开发--准备工作 3周前 (03-13)
- C#微信公众号开发--微信事件交互 3周前 (03-13)
- C#微信公众号开发--网页授权(oauth2.0)获取用户基本信息一 3周前 (03-13)