用Python写一个小型数据管理系统,要求能对数据进行删改和生成表格
时间: 2023-05-31 22:04:40 浏览: 88
以下是一个Python小型数据管理系统的示例代码,使用了Pandas库进行数据管理和生成表格:
```python
import pandas as pd
# 读取数据文件
data = pd.read_csv('data.csv')
# 定义函数:添加数据
def add_data():
name = input('请输入姓名:')
age = input('请输入年龄:')
gender = input('请输入性别:')
new_data = pd.DataFrame({'姓名': [name], '年龄': [age], '性别': [gender]})
data = pd.concat([data, new_data], ignore_index=True)
data.to_csv('data.csv', index=False)
print('添加成功!')
# 定义函数:删除数据
def delete_data():
name = input('请输入要删除的姓名:')
delete_row = data[data['姓名'] == name].index
data.drop(delete_row, inplace=True)
data.to_csv('data.csv', index=False)
print('删除成功!')
# 定义函数:修改数据
def modify_data():
name = input('请输入要修改的姓名:')
modify_column = input('请输入要修改的属性(姓名、年龄、性别):')
modify_value = input('请输入修改后的值:')
data.loc[data['姓名'] == name, modify_column] = modify_value
data.to_csv('data.csv', index=False)
print('修改成功!')
# 定义函数:生成表格
def generate_table():
table = pd.read_csv('data.csv')
print(table)
# 主程序
while True:
print('欢迎使用数据管理系统!请选择功能:')
print('1. 添加数据')
print('2. 删除数据')
print('3. 修改数据')
print('4. 生成表格')
print('5. 退出程序')
choice = input()
if choice == '1':
add_data()
elif choice == '2':
delete_data()
elif choice == '3':
modify_data()
elif choice == '4':
generate_table()
elif choice == '5':
break
else:
print('输入错误,请重新选择功能!')
```
该系统的数据存储在一个CSV文件中,包含三个属性:姓名、年龄、性别。通过用户输入实现添加、删除、修改数据的功能,同时可生成表格展示当前数据。
阅读全文