Numpy 将条件逻辑表述为数组运算

最新推荐文章于 2023-12-19 11:22:29 发布

原创最新推荐文章于 2023-12-19 11:22:29 发布 · 895 阅读

0 ·

CC 4.0 BY-SA版权

机器学习同时被 2 个专栏收录

23 篇文章

订阅专栏

Python

8 篇文章

订阅专栏

本文介绍如何利用NumPy库中的np.where函数根据条件数组对不同数组进行选择操作，并展示了如何通过np.where函数对随机生成的数组进行阈值判断并替换元素值。

#5将条件逻辑表述为数组运算--------------
xarr = np.arange(1.1,1.6,0.1)
yarr = np.arange(2.1,2.6,0.1)
cond = np.array([True, False, True,True,False])
print xarr
print yarr
print cond
#假设我们要根据cond的值选取xarr和yarr的值：当cond中的值为True时，选取
#xarr的值，否则从yarr中选取。
result = np.where(cond,xarr,yarr)#第二个和第三个参数不必是数组，他们都可以是标量值
print result

arr = np.random.randn(4,4)
print arr
result = np.where(arr > 0, 2,-2)#将随机产生的矩阵大于0的值变为2，小于0的变为-2
print result