栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

numpy:获取二维数组最小值的列和行索引

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

numpy:获取二维数组最小值的列和行索引

您可以使用

np.where

In [9]: np.where(x == np.min(x))Out[9]: (array([2]), array([1]))

就像评论中提到的@senderle一样,要获取数组中的值,您可以使用

np.argwhere

In [21]: np.argwhere(x == np.min(x))Out[21]: array([[2, 1]])

更新:

正如OP的时间所显示的那样,并且需要更加清晰

argmin
(没有重复的分钟数等),我认为可以稍微改善OP的原始方法的一种方法是使用
divmod

divmod(x.argmin(), x.shape[1])

对它们进行计时,您会发现额外的速度点虽然不多,但仍然有所改进。

%timeit find_min_idx(x)1000000 loops, best of 3: 1.1 µs per loop%timeit divmod(x.argmin(), x.shape[1])1000000 loops, best of 3: 1.04 µs per loop

如果您真的很关心性能,可以看看 cython



转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/669568.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号