按列索引在一个整型数组的每一行中选择一个元素
是否有更好的方法从“ 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]
问题答案：
您可以从给定的数组中选择，使用numpy.choose该数组可以从索引数组（在您的情况下select_id）和一组数组（在您的情况下input_array）中构造一个数组以供选择。但是，您可能首先需要转置input_array以匹配尺寸。下面显示了一个小示例：
In [101]: input_array
Out[101]:
array([[ 3, 14],
[12, 5],
[75, 50]])
In [102]: input_array.shape
Out[102]: (3, 2)
In [103]: select_id
Out[103]: [0, 1, 1]
In [104]: output_array = np.choose(select_id, input_array.T)
In [105]: output_array
Out[105]: array([ 3, 5, 50])
-
问题内容： 我有两个一维数组x和y，一个比另一个小。我试图找到x中y的每个元素的索引。 我发现有两种简单的方法可以做到这一点，第一种很慢，第二种需要占用大量内存。 记忆猪 是否有更快的方法或更少的内存密集型方法？理想情况下，搜索将利用以下事实：我们不是在列表中搜索一件事，而是在搜索许多东西，因此稍微适合并行化。如果您不假设y的每个元素实际上都在x中，则可获得加分。 问题答案： 正如Joe King
-
我被分配了一项编程任务，但我被卡住了。其说明如下： 有一个名为“秘密圣诞老人”（给他们礼物）的游戏，有很多孩子参加。对于每个参与的孩子，都有一个来自参与孩子的秘密圣诞朋友。我必须编写一个程序，为每个参与的孩子挑选一个秘密的圣诞老人朋友。 示例：如果Bob，Alice，John和George是参与的孩子，在随机选择之后， 输出可能看起来像 具有相同输入的连续两次程序运行不应有相同的结果。 我的想法是
-
问题内容： 如标题所示，我想选择以分组的每组行的第一行GROUP BY。 具体来说，如果我有一个purchases看起来像这样的表： 我的输出： 我想查询每个人id最大的购买金额（total）customer。像这样的东西： Expected Output: 问题答案： 在PostgreSQL中，这通常更简单，更快捷（下面将进行更多性能优化）： 或更短（如果不太清楚），输出列的序号为： 如果tot
-
问题内容： 我有一个生成的DataFrame，如下所示： 结果如下： 如您所见，DataFrame按升序排列，然后按降序排列。 我想选择每个组的第一行，即 从小时== 0的组中选择（0，cat26,30.9） 从小时= 1的组中选择（1，cat67,28.5） 从小时== 2的组中选择（2，cat56,39.6） 等等 因此，所需的输出将是： 能够选择每个组的前N行也可能很方便。 任何帮助都将受到
-
问题内容： 我有一个表的列，，，使得 给我下表 我将如何获得每个x，y组的第一行？也就是说，我将如何获得下表 这是第二个示例：对于这样的表T 我期望得到 问题答案： 考虑到SQL 2005或更高版本：
-
问题内容： 我有大量的数据，需要比较该数组中所有样本的距离与该数组中所有其他元素的距离。以下是我的数据集的一个非常简单的示例。 有没有一种方法可以为非示例索引的索引建立一个numpy数组？在上面的示例中，我创建了一个名为other_indexes的列表。由于各种原因，我宁愿不必这样做（大型数据集，线程，正在运行的系统上的内存非常低等，等等）。有没有办法做类似的事情.. 我读到numpy面具可以做到