f47485cc4a21fb18564ede7ab0cc9001dbd0f1d9,chaospy/descriptives/sensitivity/main.py,,Sens_m,#,9

Before Change

    dim = len(dist)
    poly = setdim(poly, dim)

    zero = [0]*dim
    out = numpy.zeros((dim,) + poly.shape)
    V = Var(poly, dist, **kws)
    for i in range(dim):
        zero[i] = 1
        out[i] = Var(E_cond(poly, zero, dist, **kws),
                     dist, **kws)/(V+(V == 0))*(V != 0)
        zero[i] = 0
    return out

After Change

    out = numpy.zeros((dim,) + poly.shape)
    variance = Var(poly, dist, **kws)
    valids = variance != 0

    for idx, unit_vec in enumerate(numpy.eye(dim, dtype=int)):

        conditional = E_cond(poly[valids], unit_vec, dist, **kws)
        out[idx, valids] = Var(conditional, dist, **kws)
        out[idx, valids] /= variance[valids]

    return out

Instances