索引最大值,寻找最大值及其位置

文章介绍了两种使用numpy库在二维数组中找到最大值及其位置的方法。第一种利用np.max和np.argmax函数,第二种通过将数组flatten转为list再获取索引。最后还展示了如何将列表转化为数组。
摘要由CSDN通过智能技术生成
# 索引最大值

方法1:
import numpy as np
 
A = np.random.random((50,50))

max_A = np.max(A) # 最大值
 
loc_max_A = np.argmax(A) # location

loc = [loc_max_A // A.shape[1], loc_max_A % A.shape[1]]
 
print(loc)

方法2:
import numpy as np
 
A = np.random.random((50,50))

B=A.flatten().tolist() # 将array转化成list类型, 才可以index

b=B.index(np.max(B))

loc = [b // B.shape[1], b % B.shape[1]] # //表示整数除法,%表示取余数
 
print(loc)

# 将列表转化为数组
import array

# 定义列表
my_list = [1, 2, 3, 4, 5]

# 将列表转换为数组
my_array = array.array('i', my_list)
print(my_array)


评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值