Проблема с визуализацией при Python Plotly Pandas? - PullRequest
Новая
       29

Проблема с визуализацией при Python Plotly Pandas?

0 голосов
/ 19 февраля 2020

У меня Pandas фрейм данных вот так: data = pd.DataFrame({"Risk":["good", "bad", "good", "good", "bad"], "Age":[22, 50, 43, 27, 19]})

, и я хочу добиться чего-то вроде этого: enter image description here

Почему мой код делает не работа ? Не могли бы вы восстановить мой код? У меня ошибка, что: AttributeError: 'list' object has no attribute 'loc'

import plotly.graph_objects as go
import plotly.tools as tls
import chart_studio.plotly as py
df_good = data.loc[data["Risk"] == 'good']['Age'].values.tolist()
df_bad = data.loc[data["Risk"] == 'bad']['Age'].values.tolist()
df_age = data['Age'].values.tolist()


#First plot
trace0 = go.Histogram(
    x=df_good,
    histnorm='probability',
    name="Good Credit"
)
#Second plot
trace1 = go.Histogram(
    x=df_bad,
    histnorm='probability',
    name="Bad Credit"
)
#Third plot
trace2 = go.Histogram(
    x=df_age,
    histnorm='probability',
    name="Overall Age"
)

#Creating the grid
fig = tls.make_subplots(rows=2, cols=2, specs=[[{}, {}], [{'colspan': 2}, None]],
                          subplot_titles=('Good','Bad', 'General Distribuition'))

#setting the figs
fig.append_trace(trace0, 1, 1)
fig.append_trace(trace1, 1, 2)
fig.append_trace(trace2, 2, 1)

fig['layout'].update(showlegend=True, title='Age Distribuition', bargap=0.05)
py.iplot(fig, filename='custom-sized-subplot-with-subplot-titles')
...