make_zipf.py

import random
import math
import argparse


parser = argparse.ArgumentParser(description='send packets')
parser.add_argument('--dist', type=int, required=True, default=80, help='flow distribution')
parser.add_argument('--pktnum', type=int, required=True, default=0, help='number of packets')

a = parser.parse_args()

num_query = a.pktnum
path_query = "z" + str(a.dist) + "_dist_" + str(num_query / 10)
zipf = a.dist / 100.0

max_key = 4999

#Zipf
zeta = [0.0]
for i in range(1, max_key + 1):
    zeta.append(zeta[i - 1] + 1 / pow(i, zipf))
field = [0] * (num_query + 1)
k = 1
for i in range(1, num_query + 1):
    if (i > num_query * zeta[k] / zeta[max_key]):
        k = k + 1
    field[i] = k

del field[0]

with open("build/" + path_query, 'w') as f:
  for i in range(0, num_query):
    a = random.choice(field)
    num = "%04X" % a
    f.write(num + '\n')