誕生日占いを確率論で見てみよう【誕生数】

from datetime import date,timedelta

def main():
    """メイン関数"""
    count_date = date(1, 1, 1)
    counter = {
        1: 0, 2: 0, 3: 0, 4: 0, 5: 0,
        6: 0, 7: 0, 8: 0, 9: 0,
        11: 0, 22: 0, 33: 0
    }
    cont = 0

    while(True):
        year = count_date.year
        month = count_date.month
        day = count_date.day

        birthday_number = calc_birthday_number(
            number_division(year) +
            number_division(month) +
            number_division(day)
        )

        counter[birthday_number] += 1
        cont += 1

        if (year == 9999 and month == 12 and day == 31):
            break

        count_date += timedelta(days=1)

    print("num\tcnt\t%")
    for count in counter:
        print(str(count) + "\t" + str(counter[count]) + "\t" + str(round((counter[count]/cont) * 100, 2)))
    
def number_division(num):
    """数値分割
    Args:
        num (int): 数値
    Return:
        list: 分割後のリスト
    """
    return [int(x) for x in list(str(num))]

def calc_birthday_number(arr):
    """誕生数算出
    Args:
        arr (list): 数値リスト
    Returns:
        int: 誕生数
    """
    num = sum(arr)
    if ((1 <= num <= 9) or num in [11, 22, 33]):
        return num

    return calc_birthday_number(number_division(num))

if __name__ == "__main__":
    main()