有时候我们需要用到各种elif才能实现一些功能,还记得excel中有if函数可以无限嵌套实现每次的if选择。
于是numpy中的where函数也可以实现这样的功能。
下面我们用这个函数编写一个haar函数。
import numpy as np
def h_function(z, a, b):
'''
a is the window.
b is the center of function.
z is the height or altitude in this case.(numpy array or one value)
'''
logical_positive = np.logical_and(z>=b, z<=b+a/2)
logical_negative = np.logical_and(z>= b - a/2, z<b)
tmpvalue = np.where(logical_positive, 1, np.where(logical_negative, -1, 0))
return tmpvalue
如上函数可以实现返回haar函数。
