You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.9 KiB
66 lines
1.9 KiB
import os
|
|
import json
|
|
import pandas as pd
|
|
import matplotlib.pyplot as plt
|
|
import seaborn as sns
|
|
|
|
data_dir = './'
|
|
|
|
all_data = []
|
|
all_tap_logs = []
|
|
|
|
for filename in os.listdir(data_dir):
|
|
if filename.endswith('.json'):
|
|
file_path = os.path.join(data_dir, filename)
|
|
with open(file_path, encoding='utf-8') as file:
|
|
data = json.load(file)
|
|
tap_logs = data.get('sensorLog', {}).get('tapLog', [])
|
|
for entry in tap_logs:
|
|
entry['participant_id'] = filename
|
|
all_tap_logs.append(entry)
|
|
|
|
df = pd.DataFrame(all_tap_logs)
|
|
|
|
df['timestamp'] = pd.to_datetime(df['timestamp'])
|
|
df['path'] = df['url'].apply(lambda x: x.split('://')[-1].split('/', 1)[-1])
|
|
|
|
# Mapping with tour_operators being mergeable to study page 2. :)
|
|
path_to_label = {
|
|
"study-page-1": "Study Page 1",
|
|
"study-page-2": "Study Page 2",
|
|
"study-page-3": "Study Page 3",
|
|
"study-page-4": "Study Page 4",
|
|
"study-page-5": "Study Page 5",
|
|
"study-page-6": "Study Page 6",
|
|
#"tour_operators": "Study Page 2"
|
|
}
|
|
|
|
df['label'] = df['path'].map(path_to_label)
|
|
|
|
grouped = df.groupby('label')
|
|
|
|
# JSON list structure
|
|
def generate_heatmap_data(group):
|
|
heatmap_data = group[['x', 'y']].copy()
|
|
heatmap_data['radius'] = 40
|
|
heatmap_data['value'] = 5
|
|
heatmap_data['x'] = heatmap_data['x'].astype(str)
|
|
heatmap_data['y'] = heatmap_data['y'].astype(str)
|
|
heatmap_data_list = heatmap_data.to_dict(orient='records')
|
|
|
|
min_value = 1
|
|
max_value = 9999
|
|
|
|
return {
|
|
"min": min_value,
|
|
"max": max_value,
|
|
"data": heatmap_data_list
|
|
}
|
|
|
|
for label, group in grouped:
|
|
heatmap_data = generate_heatmap_data(group.apply(lambda x: round(x)))
|
|
json_filename = f"{label.replace(' ', '_').lower()}.json"
|
|
with open(json_filename, 'w', encoding='utf-8') as json_file:
|
|
json.dump(heatmap_data, json_file, indent=4)
|
|
|
|
print(f"Generated {json_filename} with {len(heatmap_data)} records.")
|