NumPy的实用函数整理之extract

1495-冯同学

发表文章数:86

首页 » 数据科学库 » 正文

NumPy函数extract()用于根据特定条件从数组中提取特定元素。

extract()

extract(condition, arr)

其中参数:

  • condition : numpy数组,元素为布尔值
  • arr:与condition相同维数的numpy数组,被抽取的输入数组

函数返回值:
返回一个1维numpy数组

代码示例:
输入:

arr = np.arange(12).reshape((3, 4))
arr

输出:

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

输入:

condition = np.mod(arr, 3)==0
condition

输出:

array([[ True, False, False,  True],
       [False, False,  True, False],
       [False,  True, False, False]])

输入:

np.extract(condition, arr)

输出:

array([0, 3, 6, 9])

拜师教育学员文章:作者:1495-冯同学, 转载或复制请以 超链接形式 并注明出处 拜师资源博客
原文地址:《NumPy的实用函数整理之extract》 发布于2022-01-17

分享到:
赞(0) 打赏

评论 抢沙发

评论前必须登录!

  注册



长按图片转发给朋友

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏

Vieu3.3主题
专业打造轻量级个人企业风格博客主题!专注于前端开发,全站响应式布局自适应模板。

登录

忘记密码 ?

您也可以使用第三方帐号快捷登录

Q Q 登 录
微 博 登 录