tests2.py 文件源码

def stup():
    x0 = x[0]
    init_cond = np.zeros_like(X)
    for i in range(0, x_nods_quantity):
        if 0 <= x0 < 0.3:
            init_cond[i] = 0
            x0 += h
        elif(0.3 <= x0 <= 0.7):
            init_cond[i] = 1
            x0 += h
        else:
            init_cond[i] = 0
            x0 += h
    return init_cond
#stupenka end
#different initial conditions end

# different transfer velocity