Создать BarGraph из DataFrame

Question

Итак, у меня есть сгенерированный фрейм данных Disease_Data, который имеет 2 столбца: Location и Data (см. Ниже).

Location    Data
Alabama     30.675
Alaska      27.3917
Arizona     25.825
Arkansas    29.2
California  29.117
Colorado    35.65

Я хотел создать гистограмму, как показано ниже:

Однако, когда я попробовал приведенный ниже код, все не сработало и выдало ошибку: KeyError: 'Location'

fig = plt.figure()
ax = fig.add_axes([0,0,1,1])
Disease_Data_Loc = Disease_Data['Location']
Disease_Data_Value = Disease_Data['Data']
ax.bar(Disease_Data_Loc ,Disease_Data_Value )
plt.show()

Пожалуйста, помогите, спасибо

Answer 1

• Просто постройте фрейм данных

import pandas as pd
import matplotlib.pyplot as plt

data = {'Location': ['Alabama', 'Alaska', 'Arizona', 'Arkansas', 'California', 'Colorado'],
        'Data': [30.675, 27.3917, 25.825, 29.2, 29.116999999999997, 35.65]}

df = pd.DataFrame(data)

# plot
df.plot('Location', 'Data', kind='bar', legend=False)
plt.ylabel('Cases')
plt.title('Cases of Disease by State')
plt.show()