写一个Python代码 把你给我的三个部分的csv全部录入
时间: 2024-12-20 14:18:56 浏览: 20
为了将您提供的关系模型数据录入到CSV文件中,我们可以编写一个Python脚本来生成这些CSV文件。假设我们有三个部分的数据:电影(`movies`)、系列剧集(`series`)和游戏(`games`)。我们将为每个表生成相应的CSV文件。
以下是一个示例Python脚本,它会生成这些CSV文件:
```python
import csv
from datetime import datetime
# 定义数据结构
data = {
'movies': [
{'movie_id': 1, 'movie_name': 'Inception', 'description': 'A thief who steals corporate secrets...', 'duration': 148, 'p_year': 2010, 'p_month': 7, 'e_year': 2010, 'e_month': 7, 'e_day': 16, 'r_year': 2010, 'r_month': 7, 'r_day': 16, 'director_id': 1},
# 添加更多电影数据
],
'series': [
{'series_id': 1, 'series_name': 'Stranger Things', 'description': 'When a young boy vanishes, a small town faces...'},
# 添加更多系列剧集数据
],
'games': [
{'game_id': 1, 'game_name': 'Fortnite', 'online': True, 'controller_support': True, 'publish_year': 2017, 'e_year': 2017, 'e_month': 7, 'e_day': 25, 'r_year': 2017, 'r_month': 7, 'r_day': 25},
# 添加更多游戏数据
]
}
# 生成CSV文件
def generate_csv(data, filename):
with open(filename, mode='w', newline='', encoding='utf-8') as file:
writer = csv.writer(file)
# 写入表头
headers = data[0].keys()
writer.writerow(headers)
# 写入数据行
for row in data:
writer.writerow(row.values())
# 调用函数生成CSV文件
generate_csv(data['movies'], 'movies.csv')
generate_csv(data['series'], 'series.csv')
generate_csv(data['games'], 'games.csv')
print("CSV files have been generated successfully.")
```
### 说明
1. **数据结构**:我们定义了一个字典 `data`,其中包含了三个部分的数据:`movies`、`series` 和 `games`。每个部分都是一个列表,列表中的每个元素是一个字典,表示一条记录。
2. **生成CSV文件**:`generate_csv` 函数接受两个参数:数据列表和文件名。该函数打开指定的文件,并使用 `csv.writer` 将数据写入文件。
3. **调用函数**:我们分别调用 `generate_csv` 函数来生成 `movies.csv`、`series.csv` 和 `games.csv` 文件。
您可以根据实际需要扩展 `data` 字典中的数据,以包含更多的记录。运行此脚本后,您将在当前目录下找到生成的CSV文件。
阅读全文