heaviside.py

import numpy as np

def heaviside(t):
    """
    Vectorized Heaviside step function: 0 for t<1, 1 for t>=1
    """
    h = np.zeros_like(t)
    h[t >= 1] = 1
    return h