小编ami*_*gha的帖子

我正在努力使用以下代码:

import numpy as np

e = np.linspace(0, 4, 10)

def g(x):
    if x > 1:
        return x
    else:
        return 0

vg = np.vectorize(g)

print(vg(e))

结果如下:

    [0 0 0 1 1 2 2 3 3 4]

我也检查了dtype.似乎vectorize函数正在从float64将类型转换为int64!