码迷,mamicode.com
首页 > 其他好文 > 详细

numpy高级函数:where与extract

时间:2019-10-03 18:28:50      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:HERE   style   print   tle   color   port   extra   int   numpy   

numpy高级函数:where与extract

1、numpy.where()函数,此函数返回数组中满足某个条件的元素的索引:

import numpy as np

x = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("x")
print(x)
y=np.where(x>5)
print(y)
print(x[y])

2、numpy.extract()函数,和where函数有一点相,不过extract函数是返回满足条件的元素,而不是元素索引,下面我们的例子返回数组x 的元素 模2等于0的元素。

condition=np.mod(x,2)==0
print(condition)
print(np.extract(condition,x))

 

numpy高级函数:where与extract

标签:HERE   style   print   tle   color   port   extra   int   numpy   

原文地址:https://www.cnblogs.com/shierlou-123/p/11620003.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!