50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
import pandas as pd
|
|
import matplotlib.pyplot as plt
|
|
import seaborn as sns
|
|
from .basePlotAnalysis import BasePlotAnalysis
|
|
from flask import current_app, url_for
|
|
|
|
from app.logging_config import get_logger
|
|
|
|
import matplotlib
|
|
matplotlib.use('Agg')
|
|
|
|
logger = get_logger()
|
|
|
|
class PlotLineActivityAllUsers(BasePlotAnalysis):
|
|
name = "Activity Line Graph (All Users)"
|
|
description = "This analysis shows the activity line graph for all users. Gneerates a downloadable PNG image."
|
|
plot_filename = "line_activity-all_users.png"
|
|
note = ""
|
|
|
|
def transform_data(self, df: pd.DataFrame) -> pd.DataFrame:
|
|
"""Transform data for the bar plot"""
|
|
df['hour'] = df['timestamp'].dt.hour
|
|
df = df[df['was_active'] == True].pivot_table(index='name', columns='hour', values='was_active', aggfunc='sum', fill_value=0)
|
|
df['total_active_minutes'] = df.sum(axis=1)
|
|
df = df.sort_values(by='total_active_minutes', ascending=False).drop('total_active_minutes', axis=1)
|
|
|
|
cumulative_sum_row = df.cumsum().iloc[-1]
|
|
df.loc['Cumulative Sum'] = cumulative_sum_row
|
|
|
|
return df
|
|
|
|
def plot_data(self, df: pd.DataFrame):
|
|
# Plot activity throughout the day for each user with the cumulative sum having a bold line in one plot
|
|
plt.figure(figsize=(12, 6))
|
|
|
|
# Plot each user's activity
|
|
for index, row in df.iterrows():
|
|
if index == 'Cumulative Sum':
|
|
plt.plot(row.index, row.values, label=index, linewidth=3, color='black') # Bold line for cumulative sum
|
|
else:
|
|
plt.plot(row.index, row.values, label=index)
|
|
|
|
# Add labels and title
|
|
plt.xlabel('Hour of Day')
|
|
plt.ylabel('Activity Count')
|
|
plt.title('User Activity Throughout the Day')
|
|
plt.legend(loc='upper left', bbox_to_anchor=(1, 1))
|
|
|
|
plt.grid(True)
|