Generate Binomial Permutations in Pandas

Question

I have some taps. There are n of them. Each tap can either be off or on. I would like to generate a dataframe of all of the possible states that my set of taps could be in. For example if

n=2

Then I would like to generate the dataframe

score 2 · Accepted Answer · edited Sep 04 '19 at 08:03

2

Use itertools.product with DataFrame constructor:

from  itertools import product

n = 2
df = pd.DataFrame(list(product(range(2), repeat=n)))

print (df)
   0  1
0  0  0
1  0  1
2  1  0
3  1  1

edited Sep 04 '19 at 08:03

R. Cox

819
8
25

answered Sep 04 '19 at 07:44

jezrael

822,522
95
1,334
1,252

That works for n=2 but not for other values as it always has only 2 columns. – R. Cox Sep 04 '19 at 07:55
@R.Cox - so you need `df = pd.DataFrame(product(range(n), repeat=n))` ? – jezrael Sep 04 '19 at 07:56
df = pd.DataFrame(list(product(range(2), repeat=n))) – R. Cox Sep 04 '19 at 07:57

Generate Binomial Permutations in Pandas

1 Answers1