Creates user data for kmeans.py (https://gist.github.com/Antoine-D/eb1f76efdea2d319e30d)

create_user_data.py

import random
import json

data = {}
data['dataset'] = list()

# add each user to the dataset
for user_i in range(200):
  new_user = {}
  new_user["id"] = str(user_i)
  new_user["data"] = list()

  liked_sites = [random.randint(1, 10) for i in range(random.randint(1, 5))]

  for site_i in range(len(liked_sites)):
    new_user["data"].append(str(liked_sites[site_i]) + ".com")

  data['dataset'].append(new_user)

json_data = json.dumps(data)
data_file = open("data.json", "w")
data_file.write(json_data)