你可以这样做:
import pandas as pd
import streamlit as st
def main():
try:
df = pd.read_csv("my_csv.csv", sep=",", index_col=False)
except FileNotFoundError as e:
df = pd.DataFrame({
"Name": ["Joe", "John", "Jane", "Mary"],
"Address1": ["A", "E", "I", "M"],
"Address2": ["B", "F", "J", "N"],
"Address3": ["C", "G", "K", "O"],
"Address4": ["D", "H", "L", "P"],
"City": ["Dublin", "Ontario", "London", "Paris"],
"Country": ["Ireland", "Canada", "England", "France"],
})
st.dataframe(df)
with st.form(key="form"):
col_to_change = st.selectbox("Column to change", df.columns)
new_col_name = st.text_input("New name", value="")
submit_button = st.form_submit_button(label='Submit')
if submit_button:
df = df.rename(columns={col_to_change: new_col_name})
st.dataframe(df)
df.to_csv("my_csv.csv", sep=",", index=False)
return
if __name__ == "__main__":
main()
它读取并保存到名为mycsv.csv 的文件中。开头可能会缺少此 csv。
这段代码让用户选择一个列和一个新名称,然后单击一个按钮来重命名数据框的列并将其保存到mycsv.csv,并使用新的列名:
with st.form(key="form"):
col_to_change = st.selectbox("Column to change", df.columns)
new_col_name = st.text_input("New name", value="")
submit_button = st.form_submit_button(label='Submit')
if submit_button:
df = df.rename(columns={col_to_change: new_col_name})