heatmap.py

import numpy as np; np.random.seed(0)
import seaborn as sns; sns.set()
import pandas as pd
df = pd.read_csv("data.csv")
# print(df)
df = df.pivot_table(values='data', index=['name'],columns=['frame'])
# print(df)
df.columns.str.strip()
cmap = sns.color_palette("Reds")
uniform_data = np.random.rand(10, 12)

ax = sns.heatmap(df, cmap='Reds', annot=True)

for i in range(1,97):
    if ('frame_' + str(i)) not in df.columns:
        df['frame_' + str(i)] = 0
df= df.reindex(sorted(df.columns,key=lambda x: int(x.split('_')[-1])), axis=1)

import matplotlib.pyplot as plt


fig, ax = plt.subplots(figsize=(96,5))
df = df.sort_values(by='name', ascending=False)
ax = sns.heatmap(df, cmap='Reds', annot=True, linewidths=1, ax=ax)