「Python」超简单!Python返回矩阵最大元素/最小值坐标,三行搞定!(更新:一行搞定!)

  • 更新

使用np.unravel_index()函数:

np.unravel_index(A.argmax(), A.shape)     # (2, 2)

原理同下

——————————————————————————————
假定矩阵A,数据类型为array

A = [[0. 1. 2. 3.]
 	 [3. 4. 5. 5.]
 	 [6. 7. 8. 8.]]
  • 返回最大值
h, w = A.shape
position = A.argmax()                         # position = 10
row, col = position // w, position % w        # row, col = 2, 2 (不是(2, 3))

说明:由于A.argmax()返回的是矩阵从上到下,从左到右遍历的索引值(第一个最大值的索引),因此返回值为一个整数而不是坐标,即便如此,我们也能根据返回的位置结合矩阵形状,根据取整(//)和取余(%)操作得到目标元素在矩阵中的二维索引值

  • 2
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值