Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to automatically size a matplotlib plot to fit the x axis labels

The data plots OK, but the rotated labels on the x axis are clipped. How do I open the plot such that everything fits?

def plot(data):
    import matplotlib.pyplot as plt
    # Uh oh. Our data is not what this logic expects. We need to break it into 2 lists
    
    plt.style.use('ggplot')

    breeds = [x[0] for x in data]
    totals = [x[1] for x in data]

    # 
    x_pos = [i for i, _ in enumerate(data)]   # Figure out where the bars will go

    plt.bar(x_pos, totals, color='green')
    plt.xlabel("Breed")
    plt.ylabel("Total Cows")
    plt.title("Total Cows by Breed")
    
    # We need to rotate the x axis labels to vertical because they are too long and they overlap
    plt.xticks(rotation = 90)
    plt.xticks(x_pos, breeds)  # x_pos matches one-to-one with breeds
    
    plt.show()
    
if __name__ == '__main__':
    data = [["brown",100],["White",200], ["Zebra",4000], ["Unknown", 4500]]
    plot(data)

enter image description here

like image 684
nicomp Avatar asked Jan 24 '26 09:01

nicomp


1 Answers

you can use tight layout :

    ...
    plt.tight_layout()
    plt.show()
like image 102
eshirvana Avatar answered Jan 26 '26 22:01

eshirvana