Need Probability function in Python

Question

I’m currently learning a couple of programming languages, but I need to make something like a function that works as the following:

Imagine you’ve a bag where you know you’ve 5 blue balls, 2 yellow balls, 1 pink ball. And you want to take one ball out of the bag and check it’s color. So the ideia is that the array should be [blue, yellow, pink] and not [blue, blue, blue…, yellow, yellow, pink]

Is it possible?

I’m still learning the basics.

See [choices](https://docs.python.org/3/library/random.html#random.choices) — Chris Charley, Nov 01 '22 at 02:39
I don't understand how taking out _one_ ball would result in an array with _three_ elements. — John Gordon, Nov 01 '22 at 02:48
Does this answer your question? [random.choice from set? python](https://stackoverflow.com/questions/15837729/random-choice-from-set-python) — Michael Tuchman, Nov 01 '22 at 02:48

score 0 · Answer 1 · answered Nov 01 '22 at 02:45

0

Example using random.choices

import random

a = ["blue", "yellow", "pink"]

print(random.choices(a, weights=[5/8, 2/8, 1/8]))

answered Nov 01 '22 at 02:45

jvx8ss

529
2
12

score 0 · Answer 2 · answered Nov 01 '22 at 02:58

You can try this and can modify any number of drawings you may need

from random import choices

colors = ['Blue', 'Yellow', 'Pink'] # All color choices
prob_dist = [0.625, 0.25, 0.125] # Probability Distribution of the colors in order
num_choice = 1 # Number of colors you want to pick at a time

draw = choices(colors, weights=prob_dist, k=num_choice)
print(draw)

Need Probability function in Python

2 Answers2