Too much padding with MatPlotLib

Question

Im doing a heatmap from a pandas dataframe with matplotlib. Everything looks good, only I have too much padding on the top of the heatmap and I don't know how to change it:

The code I am currently using:

plt.figure(num=1,figsize=(10,10))
plt.pcolor(table, cmap='Reds', vmin=0, vmax=5, edgecolors="black")
plt.tight_layout()
plt.yticks(np.arange(0.5, len(table.index), 1), table.index)
plt.xticks(np.arange(0.5, len(table.columns), 1), table.columns, rotation=45)
plt.savefig('test.png', bbox_inches='tight')
plt.show()

Is that because of yticks are strings and matplotlib is trying to make space in case I want to rotate them?

Thanks! Regars!

I can not reproduce the problem ... you can put a Minimal, Complete, and Verifiable example? — Lucas
– Lucas, Commented Jan 13, 2017 at 9:03

periphreal · Accepted Answer · 2017-01-13 08:59:01Z

1

Matplotlib tries to find a nice round number for the axis limits (e.g. 45 instead of 41) to make the plot look nicer with automated ticks. Redefining the ticks doesn't change the limits. You can manually force them to match your data exactly:

plt.gca().set_ylim((0, len(table.index)))
plt.gca().set_xlim((0, len(table.columns)))

answered Jan 13, 2017 at 8:59

periphreal

1119 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Too much padding with MatPlotLib

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related