distribution.py

def generate_question_numbers(distributions, total_number):
    
    generate_numbers = lambda S, N: map(
        lambda s: round( ( N * s ) / sum(S) ), 
        sorted(S)
    )
    
    question_numbers = []
    added_number = 0

    for i, question_number in enumerate(generate_numbers(distributions, 
                                                         total_number)):
        added_number += question_number
        
        if added_number >= total_number:
            question_number -= (added_number - total_number)

        question_numbers.append(question_number)

    return question_numbers