Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created November 30, 2020 07:45
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/ac398aa519cc581f08242439099482d7 to your computer and use it in GitHub Desktop.
Save amankharwal/ac398aa519cc581f08242439099482d7 to your computer and use it in GitHub Desktop.
best_nf_authors = df.groupby(['Author', 'Genre']).agg({'Name': 'count'}).unstack()['Name', 'Non Fiction'].sort_values(ascending=False)[:11]
best_f_authors = df.groupby(['Author', 'Genre']).agg({'Name': 'count'}).unstack()['Name', 'Fiction'].sort_values(ascending=False)[:11]
with plt.style.context('Solarize_Light2'):
fig, ax = plt.subplots(1, 2, figsize=(8,8))
ax[0].barh(y=best_nf_authors.index, width=best_nf_authors.values,
color=genre_col[0])
ax[0].invert_xaxis()
ax[0].yaxis.tick_left()
ax[0].set_xticks(np.arange(max(best_f_authors.values)+1))
ax[0].set_yticklabels(best_nf_authors.index, fontsize=12, fontweight='semibold')
ax[0].set_xlabel('Number of appreances')
ax[0].set_title('Non Fiction Authors')
ax[1].barh(y=best_f_authors.index, width=best_f_authors.values,
color=genre_col[1])
ax[1].yaxis.tick_right()
ax[1].set_xticks(np.arange(max(best_f_authors.values)+1))
ax[1].set_yticklabels(best_f_authors.index, fontsize=12, fontweight='semibold')
ax[1].set_title('Fiction Authors')
ax[1].set_xlabel('Number of appreances')
fig.legend(['Non Fiction', 'Fiction'], fontsize=12)
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment