def _occupied_cols(piece):
    """ Returns a list of which cols are occupied in a piece.

    Ex: >>> list(map(_occupied_cols, PIECE_I))
        [[0, 1, 2, 3], [2], [0, 1, 2, 3], [1]]
    """
    return [i for i, row in enumerate(zip(*piece)) if any(row)]