Построение 2 столбцов CSV с ошибкой matplotlib

Question

Я пытаюсь сделать простую гистограмму из файла CSV с двумя столбцами. Один столбец - это имена осей X, другой - фактические данные, которые будут использоваться для столбцов. CSV выглядит так:

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

Я использую этот код:

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)

Однако я получаю сообщение об ошибке

KeyError: 'count' из этой строки: df.plot.bar (x = 'count', y = 'team')

Я не понимаю, как происходит ошибка для чего-то существующего.

Answer 1

Когда вы устанавливаете count в качестве индекса, в вашем DataFrame остается только один столбец, то есть team.Не устанавливайте count в качестве индекса и переключайте порядок значений x и y для построения гистограммы

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')

Матплотлиб раствор

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