入门客AI创业平台(我带你入门,你带我飞行)
博文笔记

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

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

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

numpy

声明:该文观点仅代表作者本人,入门客AI创业平台信息发布平台仅提供信息存储空间服务,如有疑问请联系rumenke@qq.com。