酷应用

发现了一个好用到爆的数据分析利器

百家作者：AI100 2022-07-22 18:15:31

作者 | 俊欣

来源 | 关于数据分析与可视化

小编最近碰上了一个数据分析利器，可以将我们需要的数据展示在网页上，并且进行相对深度的数据分析与挖掘，所以就打算借此机会和大家分享一下。

关于streamlit-aggrid

我们知道用Streamlit模块来进行web应用的开发真的非常的方便，但是在展示表格方面则显得十分地简陋，只有两个简单的接口函数，分别是st.table(df)和st.dataframe(df)，对于字段较多的表格数据的展示非常的不友好，今天小编就来介绍一款Streamlit的插件，streamlit-aggrid，它的基础功能包括

数据排序
表格样式的调整
数据的筛选
翻页
等等

首先我们先通过pip命令下载该模块

pip install streamlit-aggrid

我们先来写一个简单的demo，看一下该模块到底能实现哪些功能，代码如下

import pandas as pd 
import streamlit as st 
from st_aggrid import AgGrid

st.set_page_config(page_title="网飞(Netflix)的电影数据分析", layout="wide") 
st.title("网飞(Netflix)的电影数据分析")
shows = pd.read_csv("netflix_titles.csv")  
AgGrid(shows)

output

我们和st.dataframe(shows)出来的结果相比，发现调用streamlit-aggrid模块展示出来的表格更加美观，如下图所示

不同方法的结果对比

当然我们还能够给数据进行排序，如下图所示

并且还可以根据指定的条件来进行数据的筛选，如下图所示

我们还可以按照自己的喜好来拖拽表格当中的每一列的数据，调整表格的顺序

更多操作

翻页

除了上面的一些基本操作之外，streamlit-aggrid模块展示出来的表格数据还支持翻页操作，代码如下

import pandas as pd
import streamlit as st
from st_aggrid import AgGrid
from st_aggrid.grid_options_builder import GridOptionsBuilder

st.set_page_config(page_title="网飞(Netflix)的电影数据分析", layout="wide")
st.title("网飞(Netflix)的电影数据分析")

shows = pd.read_csv("netflix_titles.csv")

gb = GridOptionsBuilder.from_dataframe(shows)
gb.configure_pagination()
gridOptions = gb.build()

AgGrid(shows, gridOptions=gridOptions)

output

分组统计

我们平常在Pandas模块当中用到的groupby分组统计来streamlit-aggrid模块当中也可以轻松地实现，代码如下

import pandas as pd
import streamlit as st
from st_aggrid import AgGrid
from st_aggrid.grid_options_builder import GridOptionsBuilder

st.set_page_config(page_title="网飞(Netflix)的电影数据分析", layout="wide")
st.title("网飞(Netflix)的电影数据分析")

shows = pd.read_csv("netflix_titles.csv")
gb = GridOptionsBuilder.from_dataframe(shows)

gb.configure_pagination()
gb.configure_side_bar()
gb.configure_default_column(groupable=True, value=True, enableRowGroup=True, aggFunc="sum", editable=True)
gridOptions = gb.build()

AgGrid(shows, gridOptions=gridOptions, enable_enterprise_modules=True)

这样，在表格的最左侧会出现工具栏，我们可以在其中进行进一步的操作，如下图所示

高亮表格数据

在Pandas模块当中我们可以给指定的数据高亮显示，那么同样地在streamlit-aggrid模块当中也可以实现，代码如下

shows = pd.read_csv("netflix_titles.csv")
gb = GridOptionsBuilder.from_dataframe(shows)

cellsytle_jscode = JsCode(
    """
function(params) {
    if (params.value.includes('United States')) {
        return {
            'color': 'white',
            'backgroundColor': 'red'
        }
    } else {
        return {
            'color': 'black',
            'backgroundColor': 'white'
        }
    }
};
"""
)

gb.configure_column("country", cellStyle=cellsytle_jscode)
gridOptions = gb.build()
data = AgGrid(
    shows,
    gridOptions=gridOptions,
    enable_enterprise_modules=True,
    allow_unsafe_jscode=True
)