Count of sums

[mattis]

I would like to count sum(subsets) in the following code, how can i do this? thanks.

import itertools thelist = [0, 0, 0, 1, 1, 2, 2, 1] for L in range(5,6): for subset in itertools.combinations(thelist, L): print(f"{subset} => {sum(subset)}") 

For example, I would like to count where sum(subset)=6,7 etc.

[Larz60+]

please clarify:
count increment condition(s) --> ?
it's not clear as written

[mattis]

When I run this script, I get a result that looks like this:

(0, 0, 0, 1, 1) => 2
(0, 0, 0, 1, 2) => 3
(0, 0, 0, 1, 2) => 3
(0, 0, 0, 1, 1) => 2
(0, 0, 0, 1, 2) => 3
(0, 0, 0, 1, 2) => 3
(0, 0, 0, 1, 1) => 2
(0, 0, 0, 2, 2) => 4
(0, 0, 0, 2, 1) => 3
(0, 0, 0, 2, 1) => 3
(0, 0, 1, 1, 2) => 4
(0, 0, 1, 1, 2) => 4
(0, 0, 1, 1, 1) => 3
(0, 0, 1, 2, 2) => 5
(0, 0, 1, 2, 1) => 4
(0, 0, 1, 2, 1) => 4
(0, 0, 1, 2, 2) => 5
(0, 0, 1, 2, 1) => 4
(0, 0, 1, 2, 1) => 4
(0, 0, 2, 2, 1) => 5
(0, 0, 1, 1, 2) => 4

On the right is the sums of subsets which are in parenthesis. I'd like to count numbers of say 4s, 5s, 6s etc and printed at the bottom. How can I achieve this?

[nilamo]

Counter from the collections module can handle this easily enough:

>>> from collections import Counter >>> totals = Counter() >>> items = [0, 0, 0, 1, 1, 2, 2, 1] >>> for L in range(5, 6): ... for subs in itertools.combinations(items, L): ... totals[sum(subs)] += 1 ... >>> totals Counter({4: 19, 5: 15, 3: 9, 6: 9, 2: 3, 7: 1})

[mattis]

thank you, Nilamo.

I changed it like this with your suggestion. but I couldn't get the result. I am a new learner, what am I doing wrong here? please

import itertools from collections import Counter thelist = [0, 0, 0, 1, 1, 2] for L in range(5,6): for subset in itertools.combinations(thelist, L): print(f"{subset} => {sum(subset)}") totals = Counter() totals[sum(subset)] += 1 totals

[nilamo]

Create the counter before the loop.
Increment the counter inside the loop.
Print the counter after the loop.

[mattis]

Thanks a lot! I got it now.
Here is the final one:

import itertools from collections import Counter totals = Counter() thelist = [0, 0, 0, 1, 1, 2, 2, 2] for L in range(5,6): for subset in itertools.combinations(thelist, L): print(f"{subset} => {sum(subset)}") totals[sum(subset)] += 1 totals

[Larz60+]

use a list:

import itertools max_digit = 9 # initialize counts list to 0 counts = [0] * (max_digit +1) thelist = [0, 0, 0, 1, 1, 2, 2, 1] for L in range(5,6): for subset in itertools.combinations(thelist, L): sum_subset = sum(subset) print(f"{subset} => {sum_subset}") counts[sum_subset] += 1 print('\nCount of sums:') for n in range(max_digit + 1): print(f"sum of {n}'s' is {counts[n]}")

result:

Output:
(0, 0, 0, 1, 1) => 2 (0, 0, 0, 1, 2) => 3 (0, 0, 0, 1, 2) => 3 (0, 0, 0, 1, 1) => 2 (0, 0, 0, 1, 2) => 3 (0, 0, 0, 1, 2) => 3 (0, 0, 0, 1, 1) => 2 (0, 0, 0, 2, 2) => 4 (0, 0, 0, 2, 1) => 3 (0, 0, 0, 2, 1) => 3 (0, 0, 1, 1, 2) => 4 (0, 0, 1, 1, 2) => 4 (0, 0, 1, 1, 1) => 3 (0, 0, 1, 2, 2) => 5 (0, 0, 1, 2, 1) => 4 (0, 0, 1, 2, 1) => 4 (0, 0, 1, 2, 2) => 5 (0, 0, 1, 2, 1) => 4 (0, 0, 1, 2, 1) => 4 (0, 0, 2, 2, 1) => 5 (0, 0, 1, 1, 2) => 4 (0, 0, 1, 1, 2) => 4 (0, 0, 1, 1, 1) => 3 (0, 0, 1, 2, 2) => 5 (0, 0, 1, 2, 1) => 4 (0, 0, 1, 2, 1) => 4 (0, 0, 1, 2, 2) => 5 (0, 0, 1, 2, 1) => 4 (0, 0, 1, 2, 1) => 4 (0, 0, 2, 2, 1) => 5 (0, 1, 1, 2, 2) => 6 (0, 1, 1, 2, 1) => 5 (0, 1, 1, 2, 1) => 5 (0, 1, 2, 2, 1) => 6 (0, 1, 2, 2, 1) => 6 (0, 0, 1, 1, 2) => 4 (0, 0, 1, 1, 2) => 4 (0, 0, 1, 1, 1) => 3 (0, 0, 1, 2, 2) => 5 (0, 0, 1, 2, 1) => 4 (0, 0, 1, 2, 1) => 4 (0, 0, 1, 2, 2) => 5 (0, 0, 1, 2, 1) => 4 (0, 0, 1, 2, 1) => 4 (0, 0, 2, 2, 1) => 5 (0, 1, 1, 2, 2) => 6 (0, 1, 1, 2, 1) => 5 (0, 1, 1, 2, 1) => 5 (0, 1, 2, 2, 1) => 6 (0, 1, 2, 2, 1) => 6 (0, 1, 1, 2, 2) => 6 (0, 1, 1, 2, 1) => 5 (0, 1, 1, 2, 1) => 5 (0, 1, 2, 2, 1) => 6 (0, 1, 2, 2, 1) => 6 (1, 1, 2, 2, 1) => 7 Count of sums: sum of 0's' is 0 sum of 1's' is 0 sum of 2's' is 3 sum of 3's' is 9 sum of 4's' is 19 sum of 5's' is 15 sum of 6's' is 9 sum of 7's' is 1 sum of 8's' is 0 sum of 9's' is 0

[Larz60+]

dueling posts

[mattis]

Thank you, Larz60+. I will examine your code too.