Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save pietrocolombo/880d886be96349f52a5b6c3e6d5b35ec to your computer and use it in GitHub Desktop.
Save pietrocolombo/880d886be96349f52a5b6c3e6d5b35ec to your computer and use it in GitHub Desktop.
counting = {}
for activity in df['class'].unique():
counting[activity] = {}
activity_df = df[df['class'] == activity]
n_instance = activity_df.shape[0]
for user in df['user'].unique():
counting[activity][user] = activity_df[activity_df['user'] == user].shape[0]/n_instance*100
bars = []
colors = ['#ff00cc', '#00ff33', '#0055ff', '#ff7700']
user_stats = {}
for user in ['debora', 'katia', 'wallace', 'jose_carlos']:
user_stats[user] = []
for activity in df['class'].unique():
user_stats[user].append(counting[activity][user])
d_bar = plt.bar(df['class'].unique(), user_stats['debora'], width = 0.7, color=colors[0], alpha = 0.7)
k_bar = plt.bar(df['class'].unique(), user_stats['katia'], bottom = user_stats['debora'], \
width = 0.7, color=colors[1], alpha = 0.7)
old_stats = [user_stats['debora'][i] + user_stats['katia'][i] for i in range(len(user_stats['debora']))]
w_bar = plt.bar(df['class'].unique(), user_stats['wallace'], bottom = old_stats, \
width = 0.7, color=colors[2], alpha = 0.7)
old_stats = [old_stats[i] + user_stats['wallace'][i] for i in range(len(user_stats['debora']))]
jc_bar = plt.bar(df['class'].unique(), user_stats['jose_carlos'], bottom = old_stats, \
width = 0.7, color=colors[3], alpha = 0.7)
plt.legend((d_bar, k_bar, w_bar, jc_bar), ('debora', 'katia', 'wallace', 'jose_carlos'), loc='center left', bbox_to_anchor=(1, 0.5))
plt.savefig(f'activity_by_user_distribution.png', dpi = 180)
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment