Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import itertools
- def n_flipped(in_list, n):
- for to_flip in itertools.combinations(range(len(in_list)), n):
- new_item = in_list[:]
- for flip_index in to_flip:
- # flip 0 to 1 or vice versa
- new_item[flip_index] = 1 - new_item[flip_index]
- yield new_item
- # Usage:
- all_flipped_variants = list(n_flipped([1, 0, 1], 2))
Add Comment
Please, Sign In to add comment