I am trying to make a list containing all possible variations of 1 and 0. like for example if I have just two digits I want a list like this:
[[0,0], [0,1], [1,0], [1,1]]
But if I decide to have 3 digits I want to have a list like this:
[[0,0,0], [0,0,1], [0,1,0], [0,1,1], [1,0,0], [1,0,1], [1,1,0], [1,1,1]]
Someone told me to use itertools, but I cannot get it to work the way I want.
>>> list(itertools.permutations((range(2))))
[(0, 1), (1, 0)]
>>> [list(itertools.product((range(2))))]
[[(0,), (1,)]]
Is there a way to do this? And question number two, how would i find documentation on modules like this? I am just flailing blindly here
itertools.product() can take a second argument: the length. It defaults to one, as you have seen. Simply, you can add repeat=n to your function call:
>>> list(itertools.product(range(2), repeat=3))
[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]
To find the docs, you can either use help(itertools) or just do a quick google (or whatever your search engine is) search "itertools python".
이 기사는 인터넷에서 수집됩니다. 재 인쇄 할 때 출처를 알려주십시오.
침해가 발생한 경우 연락 주시기 바랍니다[email protected] 삭제
몇 마디 만하겠습니다