python – NumPy使用索引列表选择每行的特定列索引
内容导读
互联网集市收集整理的这篇技术教程文章主要介绍了python – NumPy使用索引列表选择每行的特定列索引，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1027字，纯文字阅读大概需要2分钟。
内容图文
我正在努力选择NumPy矩阵每行的特定列.
假设我有以下矩阵,我称之为X：
[1, 2, 3]
[4, 5, 6]
[7, 8, 9]
我还有每行的列索引列表,我称之为Y：
[1, 0, 2]
我需要获取值：
[2]
[4]
[9]
我可以生成一个与X形状相同的矩阵,而不是带有索引Y的列表,其中每列都是0-1值范围内的bool / int,表示这是否是必需的列.
[0, 1, 0]
[1, 0, 0]
[0, 0, 1]
我知道这可以通过迭代数组并选择我需要的列值来完成.但是,这将在大数据阵列上频繁执行,这就是它必须尽可能快地运行的原因.
我想知道是否有更好的解决方案？
谢谢.
解决方法:
如果你有一个布尔数组,你可以根据它做直接选择：
>>> a = np.array([True, True, True, False, False])
>>> b = np.array([1,2,3,4,5])
>>> b[a]
array([1, 2, 3])
与您的初始示例一起,您可以执行以下操作：
>>> a = np.array([[1,2,3], [4,5,6], [7,8,9]])
>>> b = np.array([[False,True,False],[True,False,False],[False,False,True]])
>>> a[b]
array([2, 4, 9])
您也可以添加一个范围并对其进行直接选择,但这取决于您如何生成布尔数组以及您的代码看起来像YMMV.
>>> a = np.array([[1,2,3], [4,5,6], [7,8,9]])
>>> a[np.arange(len(a)), [1,0,2]]
array([2, 4, 9])
希望有所帮助,如果您有任何疑问,请告诉我.
内容总结
以上是互联网集市为您收集整理的python – NumPy使用索引列表选择每行的特定列索引全部内容，希望文章能够帮你解决python – NumPy使用索引列表选择每行的特定列索引所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。
内容备注
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。
内容手机端
扫描二维码推送至手机访问。