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

Posted 2022-08-02 AI科技大本营

作者 | 俊欣

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

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

关于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
)

我们将国家为“美国”的电影数据用红色高亮显示出来，如下图所示

数据集的获取链接是：https://www.kaggle.com/datasets/shivamb/netflix-shows?resource=download

往期回顾

太卷了！AI 高数考试正确率81%

马斯克称已将大脑上传云端，网友热议！

撒贝宁搭档数字人主持，还是头一回！

如何用 Python 实现景区安防系统？

分享
点收藏
点点赞
点在看