python - 为什么我的函数的输出是二进制的？

Question

我编写了一个可微的 Heaviside 函数并将其矢量化。然而，输出似乎是奇怪的和二进制的。代码如下:

import numpy as np
import matplotlib.pyplot as plt

def heaviside(x, epis):
    if (x>= epis):
        y=1
    elif (x< -epis):
        y=0;
    else:
        y = 0.5 + x/(2*epis) + np.sin(np.pi*x/epis)/(2*np.pi);

    print (x, y);
    return y;

x1 = np.linspace(-1, 1, 100);
vheaviside = np.vectorize(heaviside);
y1 = vheaviside(x1, 0.2);
plt.plot(x1, y1, 'b', alpha=0.3)
plt.show()

输入是 [-1, 1] 中的数组，输出应该是连续的 Heaviside 函数。但是，所有输出都是 0 或 1。为什么？

Answer 1

您需要指定何时vectorize它应该使用 floats 的函数:

vheaviside = np.vectorize(heaviside, [float])

否则，根据文档:

The output type is determined by evaluating the first element of the input

在本例中是一个整数。或者，通过替换例如，确保 heaviside 始终返回 float y = 1 与 y = 1.0。