Plotting 2 columns of a csv with matplotlib error

Question

I am trying to make a simple bar graph out of a 2 column CSV file. One column is the x axis names, the other column is the actual data which will be used for the bars. The CSV looks like this:

count,team
21,group1
15,group2
63,group3
22,group4
42,group5
72,group6
21,group7
23,group8
24,group9
31,group10
32,group11

I am using this code:

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

df = pd.read_csv("sampleData.csv",sep=",").set_index('count')

d = dict(zip(df.index,df.values.tolist()))

df.plot.bar(x = 'count', y = 'team')

print(d)

However, I get an error

KeyError: 'count' from this line : df.plot.bar(x = 'count', y = 'team')

I don't understand how there is an error for something that exists.

Sheldore · Accepted Answer · 2019-06-14 13:48:37Z

2

When you set the count as index, you just have a single column left in your DataFrame, i.e., team. Don't set the count as index and switch the order of x and y values for plotting the bar chart

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

df = pd.read_csv("sampleData.csv", sep=",")
df.plot.bar(x = 'team', y = 'count')

Matplotlib solution

plt.bar(df['team'], df['count']) 
plt.xticks(rotation=45) # Just rotating for better visualizaton

edited Jun 14, 2019 at 13:48

answered Jun 14, 2019 at 13:37

Sheldore

39.2k9 gold badges63 silver badges76 bronze badges

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

Collectives™ on Stack Overflow

Plotting 2 columns of a csv with matplotlib error

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related