如何在python中使用numpy获取二维矩阵(列表)中具有最多给定元素的列和行的索引

乔治·沙

例如,我有一个二维矩阵:

[0 0 0 0 5 0 0 0

 0 5 0 0 5 0 0 0

 0 0 5 5 5 5 5 0

 5 0 0 0 5 0 0 5

 0 5 0 0 0 0 5 0]

我想获取最多5s的行和列的索引。这应该是第三行和第五列。我不想使用for循环来获取它,因为它花费了太多时间。无论如何,在python中使用numpy模块来解决它?

沃伦·韦克瑟(Warren Weckesser)

这是一种方法。a是数组,elem是要检查的元素:

In [71]: a
Out[71]: 
array([[0, 0, 5, 0, 5, 0, 0, 0],
       [0, 5, 0, 0, 0, 5, 0, 5],
       [0, 0, 5, 0, 5, 0, 5, 0],
       [0, 5, 0, 0, 0, 5, 0, 5],
       [0, 0, 0, 0, 0, 5, 0, 0]])

In [72]: elem = 5

这使(第一)列的出现次数最多elem

In [73]: (a == elem).sum(axis=0).argmax()
Out[73]: 5

这使(第一行)出现次数最多的行elem

In [74]: (a == elem).sum(axis=1).argmax()
Out[74]: 1

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

确定二维矩阵中具有偶数行和列索引的条目总和时出错

给定任意尺寸的二维列表中元素的行和列,如何递归获取对角线?

在python中测量具有相同索引的二维列表元素的平均值

Python:如何获取二维数组中具有大多数元素的子数组的索引

如何在haskell中具有特定索引的二维列表中添加数字

带有numpy的Python:如何根据特定索引从二维数组的每一行中删除元素

如何获取和设置带有二维矩阵列索引的一维数组?

如何在不使用numpy的情况下获取所有对角二维列表?

如何在熊猫数据框具有非 NA 值的二维矩阵中查找所有索引

python - 如何在Python3中使用键和值数组作为二维列表从用户那里获取字典?

numpy,在给定索引列表的情况下,是否可以对矩阵(二维数组)的特定元素求和?

如何通过一维列表中的值屏蔽二维 numpy 矩阵的行?

如何从给定的索引中构成二维列表资本的元素?

如何在 Python 中更改二维列表的列?

在纯python中(没有numpy等),如何找到二维列表中某些列的均值?

如何在python中按切片获取二维数组的所有元素

如何在python中二维列表中获取一行的平均值?

二维矩阵中的python索引

如何从python中的列表创建矩阵或二维列表?

如何在numpy中以二维矩阵随机采样

在numpy二维数组中获取与零相邻的所有元素的索引

numpy:获取二维数组最小值的列和行索引

numpy二维数组中的行和列以哪种方式用作矩阵?

如何在二维数组中使用更多索引?

如何在numpy二维数组中找到与某个列表匹配的所有元素?

获取给定二维范围内具有给定值的单元格的行或列

获取二维numpy ndarray或numpy矩阵中前N个值的索引

如何在Java列明智的二维数组中获取最大和最小的元素我有行明智的输出

如何在二维数组中检查行和列中的四个元素是否相同?