21608a6b6e6883504a0bcb06abbf9c7d8029897b,ann_benchmarks/distance.py,,transform_dense_to_sparse,#,17

Before Change

    indices = np.transpose(np.where(X))
    keys = []
    l = []
    last_i = None
    for i, j in indices:
        if last_i != None and last_i != i:
            keys.append(l)
            l = []
        l.append(j)
        last_i = i
    keys.append(l)

After Change

    // get list of indices of non-zero elements
    indices = np.transpose(np.where(X))
    keys = []
    for _, js in itertools.groupby(indices, lambda ij: ij[0]):
        keys.append([j for _, j in js])

    assert len(X) == len(keys)

    return keys

Instances