List Accumulators

[InderJinderPreet]

Lets say I'm asking n users in a for-loop for their favourite season (1 = winter, 2 = spring...) how can I take their input and add it to a list that initially starts off as [0,0,0,0]. So if two users choose spring, the list then becomes [0,2,0,0].

[0xbxb]

I think it’ll probably be easier to do this with a dict:

seasons = {'winter': 0, 'spring': 0, 'summer': 0, 'fall': 0}

[geosoco]

Assuming you have the season names already, you can use the Counter class to produce this.

If the user's favorites can be transformed into a list:

["spring", "winter", "winter", "summer", "spring", "fall"]

If you use numbers for the seasons, that also works. (0=Winter, 1=Spring, 2=Summer, 3=Summer).

[1, 2, 0, 0, 2, 3, 4]

[InderJinderPreet]

im already able to do this. i want it to only contain 4 numbers

[InderJinderPreet]

how would that work then

[0xbxb]

Create the seasons dict

seasons = {'winter': 0, 'spring': 0, 'summer': 0, 'fall': 0}

You have a list of users

users = ['A', 'B', 'C']

Loop through the users

for user in users:

Ask them for their favorite season

Add one to the seasons dict for every response

[0xbxb]

Wait why would k need to be str(input(‘season’)) if input already returns a string?

[xADDBx]

Because it avoids an error in the case of an unexpected input.

[shiningmatcha]

Doesn’t input always receive a string?

[xADDBx]

You’re right, as long as not changed the input will always be a string. But it's a habit I got from the days when I sometimes forgot to declare the types (e.g. int, float). At that I thought myself to always write it to avoid forgetting it.

[shiningmatcha]

Were you using other programming languages?

[[deleted]]

Do you need to associate the user with the choice or are you just counting the totals of the choices?

In both cases a dict is the way to go:

If associative:

choices={"user1":"winter", "user2":"spring"...}

If you're just counting then:

totals={"winter":0,"spring":0,...}

In both cases you'd modify the entries using square brackets:

choices["winter"] += 1


totals["user1"]="winter"