numpy中arg**方法的应用

最新推荐文章于 2023-07-12 16:52:15 发布

weixin_44778883

最新推荐文章于 2023-07-12 16:52:15 发布

阅读量221

点赞数

CC 4.0 BY-SA版权

分类专栏：自学

本文链接：https://blog.youkuaiyun.com/weixin_44778883/article/details/90640483

自学专栏收录该内容

51 篇文章

订阅专栏

博客介绍了numpy中用于返回对象最值及排序索引的API，包括arr.argmax()、arr.argmin()和arr.argsort()，还说明了这些API可填入参数axis，axis=0按列操作，axis=1按行操作，同时提到Series可用于索引，DataFrame对象可转成ndarray格式操作。

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

numpy中返回对象中最大值，最小值或排序后的索引。
arr.argmax() # 返回array对象中最大值所占的索引
arr.argmin() # 返回array对象中最小值所在的索引
arr.argsort() # 返回array对象中升序排列后的索引
上述三个API均可填入参数axis=[0, 1, …]。
axis=0，表示按列操作
axis=1，表示按行操作
注意Series也可以用于索引
DataFrame对象可以转换成ndarray格式再进行arg***操作。

import numpy as np
import pandas as pd

data = pd.DataFrame(np.arange(24).reshape(4, 6))
data.values.argmax(axis=0)
data.values.argmin(axis=1)
data.values.argsort(axis=0)
# 也可以如下编写，效果一致
np.argmax(data.values, axis=0)
np.argsort(-data.values, axis=0)   # 降序排列