katan-haibun.py

katan-haibun.py

import random
import sys

C = ['ヒツジ', '麦', 'レンガ', '木', '石']


def katan(n):
    import collections
    c = collections.defaultdict(lambda: 0)
    for i in range(n):
        c[random.choice(C)] += 1
    return c


for k, v in katan(int(sys.argv[1])).items():
    print(f'{k} => {v}')

$ python katan-haibun.py 10
麦 => 3
レンガ => 2
木 => 4
石 => 1

requires Python >= 3.6