Aggregating and plotting multiple columns using matplotlib

Question

I've got data in a pandas dataframe that looks like this:

ID    A    B    C    D
100   0    1    0    1
101   1    1    0    1
102   0    0    0    1
...

The idea is to create a barchart plot that shows the total of each (sum of the total number of A's, B's, etc.). Something like:

      X 
  X   X
x X   X 
A B C D

This should be so simple...

mozway · Accepted Answer · 2022-11-15 13:54:40Z

4

Set 'ID' aside, sum, and plot.bar:

df.set_index('ID').sum().plot.bar()

# or
df.drop(columns=['ID']).sum().plot.bar()

output:

print(df.drop(columns='ID')
        .replace({0: ' ', 1: 'X'})
        .apply(sorted, reverse=True)
        .to_string(index=False)
     )

Output:

A B C D
X X   X
  X   X
      X

answered Nov 15, 2022 at 13:54

mozway

267k13 gold badges56 silver badges106 bronze badges

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

1 Answer 1