Python numpy，数组的布尔索引，where三元运算符，clip裁剪

 

demo.py（数组的布尔索引）：

# coding=utf-8
import numpy as np

t = np.arange(24).reshape((4,6))
print(t<10)
'''
[[ True  True  True  True  True  True]
 [ True  True  True  True False False]
 [False False False False False False]
 [False False False False False False]]
'''

t[t<10] = 0   # 将小于10的元素替换为0
print(t)
'''
[[ 0  0  0  0  0  0]
 [ 0  0  0  0 10 11]
 [12 13 14 15 16 17]
 [18 19 20 21 22 23]]
'''

demo.py（where，numpy中的三元运算符）：

# coding=utf-8
import numpy as np

t = np.arange(24).reshape((4,6))

# numpy中的三元运算符
t2 = np.where(t<10, 0, 50)  # 小于10的元素替换成0,否则替换成50
print(t2)
'''
[[ 0  0  0  0  0  0]
 [ 0  0  0  0 50 50]
 [50 50 50 50 50 50]
 [50 50 50 50 50 50]]
'''

<