牛骨文教育服务平台(让学习变的简单)
博文笔记

Numpy中找出array中最大值所对应的行和列

创建时间:2016-03-07 投稿人: 浏览次数:33265

Python特别灵活,肯定方法不止一种,这里介绍一种我觉得比较简单的方法。
如下图,使用x == np.max(x) 获得一个掩模矩阵,然后使用where方法即可返回最大值对应的行和列。
where返回一个长度为2的元组,第一个元素保存的是行号,第二个元素保存的是列号。

numpy

声明:该文观点仅代表作者本人,牛骨文系教育信息发布平台,牛骨文仅提供信息存储空间服务。