list
对于列表来说,可直接使用max()\min()或者list.index()来求得列表中最大和最小元素以及该元素的索引值:
a = [1,2,3,4,5]
print(max(a))
print(min(a))
print(a.index(max(a)))
print(a.index(min(a)))
输出:
5
1
4
0
numpy.ndarray
对于numpy的数组,可直接使用np.max()\np.min()或者np.argmax()\np.argmin()来分别求得数组中最大、最小元素及其对应的索引值。
import numpy as np
a = [1,2,3,4,5]
arr_a = np.array(a)
print(np.max(a))
print(np.min(a))
print(np.argmax(arr_a ))
print(np.argmin(arr_a ))
输出:
5
1
4
0
二维矩阵 对于二维矩阵,可以定方向(是求每行的最大值还是求每列的最大值,结果也是一个ndarray)
np.max(a,axis=0) #每列最大np.max(a,axis=1) #每行最大
附:
list和numpy array相互转化的函数:
list转array:
arr_a = np.array(a)
【信号处理|python获得list或numpy数组中最大元素及其对应的索引】array转list:
a_list = a.tolist()
推荐阅读
- paddle|动手从头实现LSTM
- 人工智能|干货!人体姿态估计与运动预测
- 推荐系统论文进阶|CTR预估 论文精读(十一)--Deep Interest Evolution Network(DIEN)
- Python专栏|数据分析的常规流程
- 读书笔记|《白话大数据和机器学习》学习笔记1
- Pytorch学习|sklearn-SVM 模型保存、交叉验证与网格搜索
- Python机器学习基础与进阶|Python机器学习--集成学习算法--XGBoost算法
- 深度学习|深度学习笔记总结
- 机器学习|机器学习Sklearn学习总结
- 机器学习|线性回归原理与python实现