How can I store a value from random.choices in python