| 123456789101112131415161718 |
- import csv
- import numpy
- debug = True
- chars = [str(d) for d in range(1, 10)]
- p = [1 for _ in chars]
- with open('letter_dist.csv', newline='') as csvfile:
- reader = csv.reader(csvfile, delimiter=',')
- sp = sum(p)
- for row in reader:
- chars.append(row[0])
- p.append(float(row[2]))
- p = numpy.array(p) / sum(p)
- def random_chars(count):
- return ''.join(numpy.random.choice(chars, p=p) for _ in range(count))
|
