numpy中常用的函数总结

努力在北京混出人样

发布于 2019-02-18 17:39:35

5970

发布于 2019-02-18 17:39:35

文章被收录于专栏：祥子的故事祥子的故事

不定期更新，多积累

where

语法格式

numpy.where(condition[, x, y])

位置函数，通过条件筛选符合要求的数据。xxx和yyy表示候选项，非必需，且两者必须同时给出或不给出。若是满足就用xxx来填充，不满足就用yyy来填充

案例展示

import numpy as np
a = np.array([[1,2,3,4],[3,4,5,6],[6,7,8,9]])
a

array([[1, 2, 3, 4],
       [3, 4, 5, 6],
       [6, 7, 8, 9]])

# 返回大于5的位置
np.where(a > 5)

(array([1, 2, 2, 2, 2]), array([3, 0, 1, 2, 3]))

# 大于5的填充为1，小于或等于5的填充为10
np.where(a > 5,1,10)

array([[10, 10, 10, 10],
       [10, 10, 10,  1],
       [ 1,  1,  1,  1]])

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2018年07月18日，如有侵权请联系 cloudcommunity@tencent.com 删除

函数