Vamos mergulhar.
Crie uma figura com linha de tendência
O primeiro passo é criar uma figura com linhas de tendência. Para este exemplo, usaremos um gráfico de dispersão simples usando os dados do TIPS, como mostrado no código abaixo:
importar plotly.expresso como px
df = px.dados.pontas()
Fig = px.dispersão (df, x = 'total_bill', y = 'tip', facet_col = 'fumante', color = 'sexo', tendline = 'ols')
Figo.mostrar()
O código acima deve retornar uma figura como mostrado:
Depois de termos a figura com linhas de tendência no rastreamento, podemos extrair os parâmetros do modelo subjacente usando a função get_trendlines_results ().
A sintaxe da função é como mostrado:
plotly.expressar.get_trendline_results (fig)
Parâmetros:
A função retornará um quadro de dados de pandas com as colunas PX_FIT_RESULTS que contêm o objeto StatsModels.
Exemplo
Vamos ver como podemos extrair as estatísticas de ajuste usando a figura que criamos anteriormente. O código é como mostrado:
resultado = px.get_trendline_results (fig)
Imprimir (resultado)
Saída:
Podemos então acessar o parâmetro do modelo, conforme mostrado na consulta abaixo:
importar plotly.expresso como px
df = px.dados.pontas()
Fig = px.dispersão (df, x = 'total_bill', y = 'tip', facet_col = 'fumante', color = 'sexo', tendline = 'ols')
resultado = px.get_trendline_results (fig)
resultado.Query ("sexo == 'masculino' e fumante == 'sim'").px_fit_results.ILOC [0].resumo()
Resultado:
Fechamento
Este artigo abrange o básico do uso da função get_trendline_results para extrair estatísticas de ajuste para linhas de tendência.