python - 根据值对数组应用不同的函数

Question

我正在尝试将不同的函数应用于 numpy 数组的第一部分和第二部分。因此，在本例中，将小于 0.5 的项乘以 2，然后将 0.5 及以上的项加 1。

def myFunc:
    x = numpy.linspace(0,1,11)
    def test():
        for i in x:
            if i < 0.5:
                a = i*2
            else:
                a = i+1.0
        return(a)

print(test())

我想返回:

[0,0.2,0.4,0.6,0.8,1.5,1.6,1.7,1.8,1.9,2]

谢谢。

Answer 1

x = np.linspace(0,1,11)
x[x<0.5]*=2
x[x>=0.5]+=1