гистограмма с использованием python matplotlib.pyplot

Question

У меня есть такой фрейм данных:

A   B   C   D   E   F   index1
44544   44544   44544   44544   44544   44544   250
0   0   0   0   761 738 500
0   0   0   0   0   13  750
0   0   0   0   1   3   1000
0   0   0   0   10  11  1250
0   0   2   0   16219   8028    1500
0   0   12560   9649    102 222 1750
0   0   26406   23089   115 56  2000

Теперь я хочу построить гистограмму с помощью переключателей в matplotlib.Я пробовал следующий код:

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.widgets import RadioButtons

df5=pd.read_excel(r'C:\YOIGO\hi.xlsx')

l1=df5.columns[0:6].tolist()
fig, ax = plt.subplots()
l,  = ax.plot(np.array(df5.index1), np.array(df5.iloc[:,2]), lw=2, color='red')
plt.subplots_adjust(left=0.3)

t=tuple(l1)
print(t)
axcolor = 'lightgoldenrodyellow'
rax = plt.axes([0.05, 0.7, 0.15, 0.15], facecolor=axcolor)
radio = RadioButtons(rax, t)
d={}
for x in t:
    d[x]=np.array(df5[x])


def hzfunc(label):
    hzdict = d
    ydata = hzdict[label]
    l.set_ydata(ydata)
    plt.draw()
radio.on_clicked(hzfunc)
plt.show()

Но приведенный выше код дает мне нормальный график, а не гистограмму.Могу ли я узнать, как преобразовать это в гистограмму?

Answer 1

Ваша проблема в том, что вы используете plt.plot(), который предназначен для рисования линейного графика, когда вы хотите использовать plt.bar() для создания гистограммы.

Однако plot() и bar() вернуть разные объекты соответственно Line2D и BarCollection.Поэтому вам нужно изменить логику в функции обратного вызова:

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.widgets import RadioButtons

d = """A   B   C   D   E   F   index1
44544   44544   44544   44544   44544   44544   250
0   0   0   0   761 738 500
0   0   0   0   0   13  750
0   0   0   0   1   3   1000
0   0   0   0   10  11  1250
0   0   2   0   16219   8028    1500
0   0   12560   9649    102 222 1750
0   0   26406   23089   115 56  2000
"""
df5=pd.read_table(StringIO(d), sep='\s+')

fig, ax = plt.subplots()
bars = ax.bar(np.array(df5.index1), height=np.array(df5['A']), color='red', width=200)
plt.subplots_adjust(left=0.3)

axcolor = 'lightgoldenrodyellow'
rax = plt.axes([0.05, 0.7, 0.15, 0.15], facecolor=axcolor)
radio = RadioButtons(rax, df5.columns[:-1])


def hzfunc(label):
    ydata = df5[label]
    for b,y in zip(bars,ydata):
        b.set_height(y)
    plt.draw()
radio.on_clicked(hzfunc)