Home > Back-end >  How to automatically size a matplotlib plot to fit the x axis labels
How to automatically size a matplotlib plot to fit the x axis labels

Time:12-18

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

CodePudding user response:

you can do use tight layout :

    ...
    plt.tight_layout()
    plt.show()
  • Related