Pandas multiple bar charts with 2 columns on X-axis

Question

I'm trying to make an array of bar charts, one chart per city with each chart displaying the counts on the Y-Axis (ranging 70 - 210) and on the X-axis I'd like to have 21 bars, one for each weekday AND time slot combination (7x3=21). This is the data

import pandas as pd
import matplotlib.pyplot as plt

data = [
    ['CITY','DAY','TIME_BIN', 'COUNT'],
    ['PHOENIX', "Friday", 1, 70],
    ['PHOENIX', "Thursday", 2, 80],
    ['PHOENIX', "Wednesday", 3, 90],
    ['ATLANTA', "Sunday", 1, 130],
    ['ATLANTA', "Monday", 2, 150],
    ['ATLANTA', "Tuesday", 3, 160],
    ['CHICAGO', "Saturday", 1, 180],
    ['CHICAGO', "Friday", 2, 200],
    ['CHICAGO', "Friday", 3, 210],
]
df = pd.DataFrame(data[1:],columns=data[0])
print(df)

          CITY        DAY  TIME_BIN  COUNT
0  PHOENIX     Friday         1     70
1  PHOENIX   Thursday         2     80
2  PHOENIX  Wednesday         3     90
3  ATLANTA     Sunday         1    130
4  ATLANTA     Monday         2    150
5  ATLANTA    Tuesday         3    160
6  CHICAGO   Saturday         1    180
7  CHICAGO     Friday         2    200
8  CHICAGO     Friday         3    210

I want the output to be some combination of the two attempts below. Combine the array functionality but with bar charts.

# Successful attempt at making an array of charts but wrong type 
df[['DAY', 'TIME_BIN']].hist(by=df['CITY'])
plt.show()

# Bar chart with proper counts but x-axis did not combine properly
ax = df.plot(x=['DAY', 'TIME_BIN'],
                                                       y='COUNT',
                                                       kind='bar',
                                                       color=["g","b"])
plt.show()

ImportanceOfBeingErnest · Accepted Answer · 2018-02-17 23:00:49Z

2

An easy solution to plot such categorical data with an additional parameter is to use searborn.

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

data = [
    ['CITY','DAY','TIME_BIN', 'COUNT'],
    ['PHOENIX', "Friday", 1, 70],
    ['PHOENIX', "Thursday", 2, 80],
    ['PHOENIX', "Wednesday", 3, 90],
    ['ATLANTA', "Sunday", 1, 130],
    ['ATLANTA', "Monday", 2, 150],
    ['ATLANTA', "Tuesday", 3, 160],
    ['CHICAGO', "Saturday", 1, 180],
    ['CHICAGO', "Friday", 2, 200],
    ['CHICAGO', "Friday", 3, 210],
]
df = pd.DataFrame(data[1:],columns=data[0])

g = sns.factorplot(x="DAY", y ='COUNT', hue='TIME_BIN', col="CITY", col_wrap=3,
                   data=df,
                   kind="bar", size=3, aspect=.8)
g.set_xticklabels(rotation=30, ha="right")
plt.tight_layout()
plt.show()

Using pandas you can use the subplots=True argument on a dataframe with several columns, this would give you one subplot per column. To this end one would first create a MultiIndex from the "DAY" and "Time_bin" column and then pivot about the "CITY" column.

import pandas as pd
import matplotlib.pyplot as plt

data = [
    ['CITY','DAY','TIME_BIN', 'COUNT'],
    ['PHOENIX', "Friday", 1, 70],
    ['PHOENIX', "Thursday", 2, 80],
    ['PHOENIX', "Wednesday", 3, 90],
    ['ATLANTA', "Sunday", 1, 130],
    ['ATLANTA', "Monday", 2, 150],
    ['ATLANTA', "Tuesday", 3, 160],
    ['CHICAGO', "Saturday", 1, 180],
    ['CHICAGO', "Friday", 2, 200],
    ['CHICAGO', "Friday", 3, 210],
]
df = pd.DataFrame(data[1:],columns=data[0])
df.set_index(['DAY','TIME_BIN'], inplace=True)

piv = df.pivot(columns="CITY").plot(kind="bar", subplots=True)
plt.tight_layout()
plt.show()

edited Feb 17, 2018 at 23:00

answered Feb 17, 2018 at 22:26

ImportanceOfBeingErnest

342k61 gold badges737 silver badges771 bronze badges

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

4 Comments

Calculus Over a year ago

Thanks that works. Is it possible that the time_bin and days of the week could both be on the x-axis or would that not make sense to do? The usage of hue= didn't occur to me until just now.

ImportanceOfBeingErnest Over a year ago

I added another option; see if that is more to your liking.

Calculus Over a year ago

Uhg, after experimenting with this, I realized what I need is a time series with TIME_BIN on x axis, each city having its own colored line and one plot for each day of the week. Should I make a new question?

ImportanceOfBeingErnest Over a year ago

Yep, apart from your personal data (which are irrelevant here) those cases seem to have nothing in common. Ask a new question if that's needed.

Collectives™ on Stack Overflow

Pandas multiple bar charts with 2 columns on X-axis

1 Answer 1

4 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

4 Comments

Your Answer

Sign up or log in

Post as a guest

Related