按列索引选择Numpy数组的每行中的一个元素
按列索引选择Numpy数组的每行中的一个元素
提问于 2013-06-13 04:10:26
有没有更好的方法从"input_array“和"select_id”中获取"output_array“？
我们能摆脱range( input_array.shape[0] )吗？
>>> input_array = numpy.array( [ [3,14], [12, 5], [75, 50] ] )
>>> select_id = [0, 1, 1]
>>> print input_array
[[ 3 14]
[12 5]
[75 50]]
>>> output_array = input_array[ range( input_array.shape[0] ), select_id ]
>>> print output_array
[ 3 5 50]页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/17074422
复制相关文章
相似问题