用Python实现数据录入、追加、数据校验并生成表格
1. 环境准备
确保你的环境中已安装了所需的库。你可以通过pip命令安装这些库：
pip install pandas openpyxl
2. 完善后的代码
import pandas as pd
import os
# 定义学生信息的列名
columns = ["姓名", "考号", "班级", "语文", "数学", "英语", "政治", "历史", "地理", "物理", "化学", "生物", "期中排名"]
# 检查并读取现有数据
def load_existing_data(filename="期末考试成绩.xlsx"):
if os.path.exists(filename):
return pd.read_excel(filename)
else:
return pd.DataFrame(columns=columns)
# 用户输入数据
def input_student_data(df):
print("请输入学生信息（输入'q'退出）：")
while True:
student_data = {
}
for col in columns:
while True:
value = input(f"请输入{col}：")
if value.lower() == 'q':
return df
try:
# 尝试将数值类型的字段转换为浮点数
if col not in ["姓名", "考号", "班级"]:
value = float(value)
if value < 0 or value > 100:
print(f"{col}的成绩必须在0到100之间，请重新输入。")
continue
student_data[col] = value
break
except ValueError:
print(f"无效的输入，请重新输入{col}：")
df.loc[len(df)] = student_data
print("学生信息已添加。")
# 生成并保存Excel文件
def save_to_excel(df, filename="期末考试成绩.xlsx"):
df.to_excel(filename, index=False)
print(f"数据已保存到 {filename}")
# 主程序
if __name__ == "__main__":
# 加载现有数据
df = load_existing_data()
# 输入新数据
df = input_student_data(df)
# 保存数据
save_to_excel(df)
3. 代码解析
3.1 导入必要的库
import pandas as pd
import os
这里导入了数据处理相关的pandas库和用于文件操作的os库。
3.2 定义学生信息的列名
columns = ["姓名", "考号", "班级", "语文", "数学", "英语", "政治", "历史", "地理", "物理", "化学", "生物", "期中排名"]
定义了一个包含所有列名的列表。
3.3 检查并读取现有数据
def load_existing_data(filename="期末考试成绩.xlsx"):
if os.path.exists(filename):
return pd.read_excel(filename)
else:
return pd.DataFrame(columns=columns)
- 检查指定的Excel文件是否存在。
- 如果文件存在，读取并返回其内容。
- 如果文件不存在，返回一个空的DataFrame，列名为预定义的列名。
3.4 用户输入数据
def input_student_data(df):
print("请输入学生信息（输入'q'退出）：")
while True:
student_data = {
}
for col in columns:
while True:
value = input(f"请输入{col}：")
if value.lower() == 'q':
return df
try:
# 尝试将数值类型的字段转换为浮点数
if col not in ["姓名", "考号", "班级"]:
value = float(value)
if value < 0 or value > 100:
print(f"{col}的成绩必须在0到100之间，请重新输入。")
continue
student_data[col] = value
break
except ValueError:
print(f"无效的输入，请重新输入{col}：")
df.loc[len(df)] = student_data
print("学生信息已添加。")
- 提示用户输入学生信息。
- 使用一个无限循环来持续接收用户输入，直到用户输入
q退出。 - 对于每个字段，尝试将输入值转换为适当的类型（字符串或浮点数）。
- 如果输入无效或成绩不在0到100之间，提示用户重新输入。
- 将输入的数据添加到DataFrame中。
3.5 生成并保存Excel文件
def save_to_excel(df, filename="期末考试成绩.xlsx"):
df.to_excel(filename, index=False)
print(f"数据已保存到 {filename}")
- 使用
pandas的to_excel方法将DataFrame保存为Excel文件。 index=False表示不保存行索引。
3.6 主程序
if __name__ == "__main__":
# 加载现有数据
df = load_existing_data()
# 输入新数据
df = input_student_data(df)
# 保存数据
save_to_excel(df)
- 调用
load_existing_data函数加载现有数据。 - 调用
input_student_data函数获取用户输入的新数据。 - 调用
save_to_excel函数将数据保存到Excel文件中。
4. 运行代码
运行上述代码后，程序会提示你逐个输入学生的各项信息。输入完成后，数据将被追加到现有的Excel文件中，如果文件不存在则会创建一个新的文件并保存数据。同时，程序会对各科成绩进行验证，确保成绩在0到100之间。
希望这个示例能帮助你实现数据追加录入并生成电子表格文件，并确保数据的有效性。
