找出numpy array数组的最值及其索引
分类:
Numpy
在list列表中,max(list)可以得到list的最大值,list.index(max(list))可以得到最大值对应的索引
但在numpy中的array没有index方法,取而代之的是where,其又是list没有的
首先我们可以得到array在全局和每行每列的最大值(最小值同理)
1 2 3 4 5 6 7 8 9 10 11 12 | a = np.arange( 9 ).reshape(( 3 , 3 )) a array([[ 0 , 1 , 2 ], [ 9 , 4 , 5 ], [ 6 , 7 , 8 ]]) print (np. max (a)) #全局最大 8 print (np. max (a,axis = 0 )) #每列最大 [ 6 7 8 ] print (np. max (a,axis = 1 )) #每行最大 [ 2 5 8 ] |
然后用where得到最大值的索引,返回值中,前面的array对应行数,后者对应列数
1 2 3 4 | print (np.where(a = = np. max (a))) (array([ 2 ], dtype = int64), array([ 2 ], dtype = int64)) print (np.where(a = = np. max (a,axis = 0 ))) (array([ 2 , 2 , 2 ], dtype = int64), array([ 0 , 1 , 2 ], dtype = int64)) |
如果array中有相同的最大值,where会将其位置全部给出
1 2 3 4 5 6 7 | a[ 1 , 0 ] = 8 a array([[ 0 , 1 , 2 ], [ 8 , 4 , 5 ], [ 6 , 7 , 8 ]]) print (np.where(a = = np. max (a))) (array([ 1 , 2 ], dtype = int64), array([ 0 , 2 ], dtype = int64)) |
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 使用C#创建一个MCP客户端
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 按钮权限的设计及实现