如何使用Python编程来创建一个包含大王、小李和小鹿手机号及QQ号的'通信录.csv'文件,并随后编写一个函数以便能够查询大王的手机号和QQ号?
时间: 2024-11-11 14:30:23 浏览: 14
首先,你可以使用Python的`csv`模块来创建一个名为'通信录.csv'的CSV文件,它是一个简单的文本文件,用于存储表格数据。这里我们将数据存储在一个字典列表中,因为字典可以方便地关联名字和电话号码。
```python
import csv
# 定义通信录的数据
contacts = [
{"name": "大王", "phone": "13812345678", "qq": "1234567890"},
{"name": "小李", "phone": "13923456789", "qq": "2345678901"},
{"name": "小鹿", "phone": "13734567890", "qq": "3456789012"}
]
# 写入CSV文件
with open('通信录.csv', 'w', newline='', encoding='utf-8') as file:
fieldnames = ["name", "phone", "qq"]
writer = csv.DictWriter(file, fieldnames=fieldnames)
writer.writeheader() # 写入表头
writer.writerows(contacts) # 写入数据
# 创建查询函数
def find_contact_info(name):
with open('通信录.csv', 'r', encoding='utf-8') as file:
reader = csv.DictReader(file)
for row in reader:
if row['name'] == name:
return {'phone': row['phone'], 'qq': row['qq']}
return None # 如果没找到匹配的名字,则返回None
# 查询大王的信息
contact_dict = find_contact_info('大王')
if contact_dict:
print(f"大王的手机号码:{contact_dict['phone']}")
print(f"大王的QQ号:{contact_dict['qq']}")
else:
print("大王的信息未找到")
```
在这个例子中,我们创建了一个CSV文件并存储了联系人信息,然后定义了一个函数`find_contact_info`用于查找特定联系人的信息。
阅读全文