streamlit文件上传
时间: 2024-09-09 13:04:37 浏览: 150
Streamlit 是一个用于创建数据应用的开源 Python 库。在 Streamlit 中,可以非常方便地实现文件上传功能,以便用户能够上传文件到你的应用中。以下是一个简单的 Streamlit 文件上传的示例代码:
```python
import streamlit as st
# 文件上传组件
uploaded_file = st.file_uploader("上传文件", type=["csv", "xlsx"])
if uploaded_file is not None:
# 文件不为空时的处理逻辑
# 这里可以进行文件的读取操作,比如读取CSV或Excel文件
file_extension = uploaded_file.name.split(".")[-1]
if file_extension in ['csv', 'xlsx']:
if file_extension == 'csv':
# 读取CSV文件的代码
import pandas as pd
data = pd.read_csv(uploaded_file)
st.write("读取CSV文件成功:", data.head())
elif file_extension == 'xlsx':
# 读取Excel文件的代码
import pandas as pd
data = pd.read_excel(uploaded_file)
st.write("读取Excel文件成功:", data.head())
else:
st.write("只支持上传 CSV 或 Excel 文件。")
else:
st.write("请上传一个文件。")
```
在这个示例中,使用了 Streamlit 的 `file_uploader` 组件来创建一个文件上传按钮。用户可以通过这个按钮上传文件,上传后,可以通过文件的扩展名来决定如何处理这个文件。例如,如果文件是 CSV 或 Excel 格式的,可以用 pandas 库来读取文件内容,并将其展示给用户。如果不支持的文件格式,会提示用户只支持 CSV 和 Excel 文件的上传。
阅读全文