Python labeled_balls_in_labeled_boxes Examples

Example #1

File: iteration.py Project: SahanGH/psi4public

def I_llbar(nballs, nboxes, labels):
    if nballs != 0:
        for idxs in labeled_balls_in_labeled_boxes(nballs, [nballs]*nboxes):
            if not all(len(i) > 0 for i in idxs):
                continue
            else:
                yield tuple(tuple(labels[i] for i in subset) for subset in idxs)

Example #2

def I_llbar(nballs, nboxes, labels):
    if nballs != 0:
        for idxs in labeled_balls_in_labeled_boxes(nballs, [nballs] * nboxes):
            if not all(len(i) > 0 for i in idxs):
                continue
            else:
                yield tuple(
                    tuple(labels[i] for i in subset) for subset in idxs)

Example #3

File: iteration.py Project: SahanGH/psi4public

def I_ll(nballs, nboxes, labels):
    if nballs == 0:
        yield (tuple(),) * nboxes
    else:
        for idxs in labeled_balls_in_labeled_boxes(nballs, [nballs]*nboxes):
            yield tuple(tuple(labels[i] for i in subset) for subset in idxs)

Example #4

def I_ll(nballs, nboxes, labels):
    if nballs == 0:
        yield (tuple(), ) * nboxes
    else:
        for idxs in labeled_balls_in_labeled_boxes(nballs, [nballs] * nboxes):
            yield tuple(tuple(labels[i] for i in subset) for subset in idxs)