Início / coding / Perguntas / 76946997
Accepted
Justin White
Asked: 2023-08-22 00:03:55 +0800 CST

percorrendo colunas para gerar countplot () seaborn

  • 772

Estou tentando criar uma função que possa reutilizar para percorrer uma lista de colunas categóricas para qualquer quadro de dados do pandas e gerar uma série de gráficos para obter visualmente o valor value_counts()de cada coluna.

Eu escrevi uma função para isso

def get_count_plot(cols_list):
    for col in cols_list:
        fig = sns.countplot(data=df[col], x=df[col].value_counts())
    return fig

Esta função retorna apenas um gráfico de uma lista de várias colunas. Não recebo erros ao concluir a função.

Eu tentei adicionar enumerate()a esta função sem sorte. Eu costumava enumeratefazer isso, mas com uma lista de dataframes em vez de uma lista de nomes de colunas.

def generate_box_plot(list_of_dfs):
    for idx, df in enumerate(list_of_dfs):
        fig = px.box(df, x='Year', y='Transportation_Cnt', color='Month')
        fig.update_xaxes(tickmode='linear')
        fig.show()

Este script funcionou muito bem, porém não estou trabalhando com uma lista de pandasdataframes.

pandas

1 respostas

  1. Best Answer

    Aqui está a alteraçãoget_count_plot()

    import seaborn as sns
import matplotlib.pyplot as plt

def get_count_plots(df, cols_list):
    for col in cols_list:
        plt.figure(figsize=(8, 6))
        sns.countplot(data=df, x=col)
        plt.title(f'Value Counts of {col}')
        plt.xticks(rotation=45)
        plt.show()

    E esta é a alteraçãoget_count_plot()

    import plotly.express as px

def generate_box_plot(dataframe, x_col, y_col, color_col):
    fig = px.box(dataframe, x=x_col, y=y_col, color=color_col)
    fig.update_xaxes(tickmode='linear')
    fig.show()

    Espero que você saiba usá-los e espero que eles funcionem para você conforme o esperado.

    • 0

relate perguntas