# -*- coding: utf-8 -*-
"""
Created on Mon Nov 28 16:04:57 2022

@author: janmo
"""
 
import random  
import pandas as pd
   
def create_lottery(p1, v1, v2):
    
    parameter_list = [p1, v1, v2]    
    expected_value = p1*v1+(1-p1)*v2
    
    if (v1 != 1) and (v2 != 1):    
        lottery = f"Mit einer Wahrscheinlichkeit von {p1*100}% erhalten Sie {v1} Punkte. Mit einer Wahrscheinlichkeit von {(1-p1)*100}% erhalten Sie {v2} Punkte."
    elif (v1 == 1) and (v2 != 1):    
        lottery = f"Mit einer Wahrscheinlichkeit von {p1*100}% erhalten Sie {v1} Punkt. Mit einer Wahrscheinlichkeit von {(1-p1)*100}% erhalten Sie {v2} Punkte."
    elif (v1 != 1) and (v2 == 1):    
        lottery = f"Mit einer Wahrscheinlichkeit von {p1*100}% erhalten Sie {v1} Punkte. Mit einer Wahrscheinlichkeit von {(1-p1)*100}% erhalten Sie {v2} Punkt." 
      
    return lottery, parameter_list, expected_value



lottery_index_list = []
lottery_text_list   = []
parameter_list      = []
expected_value_list = []

lottery_list = []

for i in range(1,51):
    
    p1_rand = round(random.random(), 1)
    v1_rand = random.randint(0,50)
    v2_rand = random.randint(0,50)
    
    lottery_creation_results = create_lottery(p1_rand, v1_rand, v2_rand)
    
    lottery_dict = dict(ID=i,
                        text=lottery_creation_results[0],
                        parameter=lottery_creation_results[1],
                        exp_value=lottery_creation_results[2])
    lottery_list.append(lottery_dict)
    

    
    # ----
    
    lottery_index_list.append(i)
    
    p1_rand = round(random.random(), 1)
    v1_rand = random.randint(0,50)
    v2_rand = random.randint(0,50)
    
    lottery_creation_results = create_lottery(p1_rand, v1_rand, v2_rand)
    
    lottery_text_list.append(lottery_creation_results[0])
    parameter_list.append(lottery_creation_results[1])
    expected_value_list.append(lottery_creation_results[2])

Lottery_df = pd.DataFrame(list(zip(lottery_index_list, lottery_text_list, parameter_list, expected_value_list)), 
                          columns=['Number', 'Text', 'Parameter', 'Expected value'])
Lottery_df['Text'][0]