TopicsBarStatic.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import json



jsonFile = open("words_per_topic.json", "r") 
words_per_topic = json.load(jsonFile)
jsonFile.close()

COLORS = {
1: '#4885ed',
2: "#db3236",
3: "f4c20d",
4: "4885ed",
5: "3cba54"
}

# Sort lists in descending order
g = []
for k in words_per_topic.keys(): 
    l = words_per_topic[k]
    sorted_list = sorted(l, key=lambda x: x[1],reverse=True)
    g.append(sorted_list)

# Get 5 most frequent words per topic
new_g = []
for item in g:
    new_g.append(item[:5])

# Update dictionary with topics names
new_dic = {}
i = 0
for k in words_per_topic.keys():
    new_dic.update({k:new_g[i]})
    i += 1
    

def getValuesLabels(k):
    labels = []
    values = []
    for pair in k:
        labels.append(pair[0])
        values.append(pair[1])
    
        
    return values, labels


def wordsTopicData():
    data = []
    colors = []
    for i in range(1,6):
        colors.append(COLORS[i])

    for k in words_per_topic.keys():
        values, labels = getValuesLabels(new_dic[k])
        values.reverse()
        labels.reverse()
        colors.reverse()
        d = {
        'type': 'bar',
        'x': values,
        'y': labels,
        'orientation': 'h',
        'marker': {
            'color': colors
        }
        }
        
        
        data.append([d])
    return data


wordsTopicData()