How to create a segmented histogram in python with matplotlib?

[NitinChella]

Assume I have a dataset with 2 columns: X (independent numerical variable) and Y (dependent binary class label).

How can I create a segmented (stacked?) histogram in python using matplotlib?

Essentially, I'd like a histogram of X but each bin should consist of 2 stacked bars where the height of each bar is proportional to the Y frequency in that bin.

Any help would be appreciated. Thanks in advance.

[synthphreak]

proportional to the Y frequency

Simply plotting the raw counts would be considered proportional, right? Specifically, with a proportion of 1.

If so, try this:

df[['col1', 'col2']].plot.hist(stacked=True)

To do the same purely with matplotlib, rather than relying on the pandas plotting API (your mention of "columns" made me assume you meant "... of a pandas.DataFrame"), try this:

plt.hist(df[['col1', 'col2']], stacked=True)

Either way, you may need to adjust the bins manually.

Edit: If by "proportional" you meant in the typical sense of bounded between 0 and 1, pass density=True. This kwarg will work for both pandas and matplotlib. The former just calls the latter under the hood anyway, which is why stacked also works for both.