主页 / coding / 问题 / 76946997
Accepted
Justin White
Asked: 2023-08-22 00:03:55 +0800 CST

循环遍历列以生成 countplot() seaborn

  • 772

我正在尝试创建一个函数，我可以重用它来循环遍历任何给定 pandas 数据框的分类列列表，并生成一系列绘图以直观地获取value_counts()每列的数据。

我为此编写了一个函数

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

此函数仅从多列列表中返回一个图。完成该功能后我没有收到任何错误。

我尝试添加enumerate()此功能但没有成功。我曾经enumerate这样做过，但使用数据框列表而不是列名列表。

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()

这个脚本工作得很好，但是我不使用数据帧列表pandas。

pandas

1 个回答

  1. Best Answer

    这里是修改后的get_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()

    这是修改后的get_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()

    我希望您知道如何使用它们，并希望它们能按预期为您工作。

    • 0

相关问题