Python使用Streamlit快速构建一个数据应用程序_Python

引言

在数据科学和机器学习领域，构建数据应用程序通常意味着需要投入大量的时间和精力来开发用户界面（ui）和后端逻辑。然而，随着streamlit的出现，这一过程变得前所未有的简单和高效。streamlit是一个开源的python库，它允许数据科学家和开发人员快速创建和分享数据应用程序，而无需具备复杂的web开发经验。本文将介绍streamlit的基本用法，并通过一个实际案例展示如何快速构建一个简单的数据应用程序。

一、streamlit简介

streamlit是一个专为数据科学家设计的工具，它允许你使用python代码来构建交互式web应用程序。通过streamlit，你可以轻松地将数据分析、机器学习模型和可视化结果嵌入到一个直观的界面中，供其他人查看和交互。streamlit的核心优势在于其易用性和灵活性，它使得数据科学家能够专注于数据本身，而不是花费大量时间在ui设计上。

二、streamlit的基本用法

在使用streamlit之前，你需要确保已经安装了python和pip。然后，你可以通过pip安装streamlit：

pip install streamlit

安装完成后，你可以通过以下命令启动一个streamlit应用程序：

streamlit hello

这个命令会启动一个默认的streamlit应用程序，展示一些基本的streamlit功能。

要创建自己的streamlit应用程序，你需要编写一个python脚本，并使用streamlit提供的api来构建ui。下面是一个简单的示例，展示如何在streamlit中创建一个带有文本输入框和按钮的界面：

import streamlit as st
 
# 设置标题
st.title('我的第一个streamlit应用程序')
 
# 创建一个文本输入框
user_input = st.text_input('请输入一些文本:')
 
# 创建一个按钮
if st.button('点击我'):
    st.write('你点击了按钮！输入的文本是：', user_input)

将上述代码保存为一个python文件（例如app.py），然后在命令行中运行以下命令来启动streamlit应用程序：

streamlit run app.py

打开浏览器并访问显示的本地地址（通常是http://localhost:8501），你将看到一个带有文本输入框和按钮的web界面。

三、streamlit的进阶用法

除了基本的文本输入框和按钮外，streamlit还支持多种ui组件，包括复选框、单选按钮、滑块、文件上传等。此外，streamlit还可以与pandas dataframe、matplotlib图表、plotly图表等常见的数据分析和可视化工具无缝集成。

1. 使用pandas dataframe

streamlit可以很方便地展示pandas dataframe，并支持对其进行筛选、排序等操作。以下是一个示例：

import streamlit as st
import pandas as pd
 
# 创建一个示例dataframe
data = {
    '姓名': ['张三', '李四', '王五'],
    '年龄': [23, 28, 22],
    '城市': ['北京', '上海', '广州']
}
df = pd.dataframe(data)
 
# 设置标题
st.title('dataframe示例')
 
# 展示dataframe
st.write(df)

2. 使用matplotlib和plotly进行可视化

streamlit支持与matplotlib和plotly等可视化库集成，使得数据可视化变得非常简单。以下是一个使用matplotlib绘制折线图的示例：

import streamlit as st
import matplotlib.pyplot as plt
import numpy as np
 
# 设置标题
st.title('matplotlib折线图示例')
 
# 生成数据
x = np.linspace(0, 10, 100)
y = np.sin(x)
 
# 绘制折线图
plt.plot(x, y)
st.pyplot(fig=plt.gcf())  # 使用st.pyplot展示图表

同样地，你也可以使用plotly来创建更复杂的可视化图表：

import streamlit as st
import plotly.express as px
import pandas as pd
 
# 创建一个示例dataframe
data = {
    '日期': pd.date_range(start='2023-01-01', periods=100),
    '值': np.random.randn(100).cumsum()
}
df = pd.dataframe(data)
 
# 设置标题
st.title('plotly折线图示例')
 
# 使用plotly绘制折线图
fig = px.line(df, x='日期', y='值')
st.plotly_chart(fig)

3. 创建交互式控件

streamlit还支持创建交互式控件，如滑块、复选框、单选按钮等，这些控件可以与你的数据分析和可视化代码进行交互。以下是一个使用滑块来控制折线图x轴范围的示例：

import streamlit as st
import matplotlib.pyplot as plt
import numpy as np
 
# 设置标题
st.title('交互式滑块示例')
 
# 生成数据
x = np.linspace(0, 10, 100)
y = np.sin(x)
 
# 创建一个滑块来控制x轴范围
slider_value = st.slider('选择x轴范围', 0.0, 10.0, (0.0, 10.0))
 
# 根据滑块值截取数据
x_selected = x[np.logical_and(x >= slider_value[0], x <= slider_value[1])]
y_selected = y[np.logical_and(x >= slider_value[0], x <= slider_value[1])]
 
# 绘制折线图
plt.plot(x_selected, y_selected)
st.pyplot(fig=plt.gcf())

四、实际案例：构建一个简单的数据应用程序

下面，我们将通过一个实际案例来展示如何使用streamlit构建一个简单的数据应用程序。这个应用程序将允许用户上传一个csv文件，并在界面上展示数据的统计摘要和可视化图表。

首先，你需要编写一个python脚本（例如data_app.py），并包含以下代码：

import streamlit as st
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
 
# 设置标题和页面布局
st.title('数据应用程序示例')
col1, col2 = st.beta_columns(2)
 
# 上传csv文件
uploaded_file = col1.file_uploader("上传csv文件", type=["csv"])
 
if uploaded_file is not none:
    # 读取csv文件
    df = pd.read_csv(uploaded_file)
    
    # 展示数据的前几行
    st.write("数据预览：")
    st.dataframe(df.head())
    
    # 展示数据的统计摘要
    st.write("数据统计摘要：")
    st.write(df.describe())
    
    # 绘制数据可视化图表
    col2.subheader("数据可视化")
    
    # 选择绘图列
    x_col = st.selectbox("选择x轴列", df.columns)
    y_col = st.selectbox("选择y轴列", df.columns)
    
    # 绘制散点图
    plt.figure(figsize=(8, 6))
    sns.scatterplot(data=df, x=x_col, y=y_col)
    st.pyplot(fig=plt.gcf())

然后，在命令行中运行以下命令来启动streamlit应用程序：