不使用pandas库是否可以重新排列/重塑CSV文件?
创始人
2024-12-29 05:01:46
0次

是的,可以使用 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 参数设置为空

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...