numpy数组的max、min、argmax和argmin计算方法

是七叔呀

已于 2024-01-22 11:55:39 修改

阅读量1k

点赞数 9

CC 4.0 BY-SA版权

文章标签： python numpy

于 2024-01-20 17:39:08 首次发布

本文链接：https://blog.csdn.net/m0_46378271/article/details/135718985

本文详细介绍了如何在Python的numpy库中使用max、min、argmax和argmin函数进行数组操作，包括轴向选择对结果的影响，并通过代码示例展示功能应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

@numpy数组的max、min、argmax和argmin计算方法

官方对numpy.max和numpy.min的说明

numpy.max
在这里插入图片描述

参考官方的理解

数组：


2	4	6	1
1	5	2	9

import numpy as np

if __name__ == '__main__':
    a = np.array([[2, 4, 6, 1], [1, 5, 2, 9]])
    print(a)
    print(np.argmax(a, axis=0))  # axis=0是指把第零个维度压缩为1, 操作(取大小、取最大等)的元素和softmax(dim=0一样,都是同一个位置的元素)
    print(np.argmax(a, axis=1))  # axis=1是指把第一个维度压缩为1
    # 预估axis=0会输出[0 1 0 1] 
    pass