我正在寻找一种矢量化的方法来索引numpy.array的numpy.array索引.
例如:
import numpy as np a = np.array([[0,3,4],[5,6,0],[0,1,9]]) inds = np.array([[0,1],[1,2],2]])
我想构建一个新数组,使得该数组中的每一行(i)都是数组a的一行(i),由数组inds(i)的行索引.我想要的输出是:
array([[ 0.,3.],# a[0][:,1]] [ 6.,0.],# a[1][:,2]] [ 0.,9.]]) # a[2][:,2]]
我可以用循环实现这个目的:
def loop_way(my_array,my_indices): new_array = np.empty(my_indices.shape) for i in xrange(len(my_indices)): new_array[i,:] = my_array[i][:,my_indices[i]] return new_array
但我正在寻找一种纯粹的矢量化解决方案.
解决方法
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。