条件逻辑运算
numpy使用where函数实现数组的条件选择和替换
where(cond, arr1, arr2) cond中元素为真时，返回arr1数组中元素，否则返回arr2数组中元素
<code>arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([5, 6, 7, 8])
cond = np.array([True, False, False, True])
rssult = np.where(cond, arr1, arr2)
# arr1=[1, 2, 3, 4]
# arr2=[5, 6, 7, 8]
# cond=True, False, False, True[]/<code>
条件替换
<code>arr = np.random.randn(4, 4)
result = np.where(arr>0, 1, -1)/<code>
- 当arr中的元素大于1时，用1替换，小于等于0时，用-1替换，返回数组只有1和-1两个元素
where多条件判断
where(cond1, value1, where(cond2, value2, value3))。cond1中元素为真时，返回value1 ，为假时判断cond2，若cond2为真返回value2，为假返回value3
<code>arr = np.random.randn(4, 4)
result = np.where(arr>10, 1, np.where(arr>5, 2, 0))/<code>
分享到:
閱讀更多 出水菱角 的文章
