是的,可以使用 Python 的内置模块 csv 来处理 CSV 文件。以下是一个示例,展示如何使用 csv 模块以及其他 Python 的标准库来重新排列/重塑 CSV 文件。
假设我们有一个名为 data.csv 的 CSV 文件,其中包含以下数据:
name,age,country
Alice,25,USA
Bob,30,Canada
Charlie,35,UK
我们想要将数据按照国家排序,重塑为以下格式:
Country,Alice,Bob,Charlie
USA,25,,,
Canada,,,30,
UK,,,35
要实现这一点,我们可以使用以下代码:
import csv
from collections import defaultdict
# 导入 CSV 文件中的数据
data = defaultdict(dict) # 使用 defaultdict 初始化嵌套字典
with open('data.csv', 'r') as f:
reader = csv.DictReader(f)
for row in reader:
# 使用国家作为外层键,名字和年龄作为内层键
data[row['country']][row['name']] = row['age']
# 重新排列并写出 CSV 文件
with open('reshaped_data.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerow(['Country', 'Alice', 'Bob', 'Charlie'])
for country, info in data.items():
# 使用默认值(空字符串)填充缺失的值
writer.writerow([country, info.get('Alice', ''), info.get('Bob', ''), info.get('Charlie', '')])
这段代码首先使用 csv.DictReader
导入 CSV 文件中的数据,并将国家作为外层字典的键,将人名和年龄作为内层字典的键。接着,将数据重新排列,再使用 csv.writer
将其写出为另一个 CSV 文件。
备注:在 Python 3 中,需要将 csv.writer
中的 newline
参数设置为空