一个关于同时读取两个CSV文件并输出的有趣案例

一个关于同时读取两个CSV文件并输出的有趣案例

代码如下所示:
import csv
with open('tets.csv', 'r') as f:
    with open('test.csv', 'r') as file:
        FileOneReader = csv.DictReader(f)
        for item, row in enumerate(FileOneReader):
            FileTwoReader = csv.DictReader(file)
            for item2, row2 in enumerate(FileTwoReader):
                print(item, row)
                print(item2, row2)
test.csv
idnameagegender
1Roberta39M
2Oliver25M
3Shayna18F
4Fechin18M
tets.csv
idnameagegender
1Roberta39M
2Oliver25M
3Shayna18F
4Fechin18M
结果:
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
1 {'id': '2', 'name': 'Oliver', 'age': '25', 'gender': 'M'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
2 {'id': '3', 'name': 'Shayna', 'age': '18', 'gender': 'F'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
3 {'id': '4', 'name': 'Fechin', 'age': '18', 'gender': 'M'}

这个情况是由于内循环for item2, row2 in enumerate(FileTwoReader):中的enumerate(FileTwoReader)其实是一个迭代器,在第一次外循环完成过后,内循环中的迭代器已经完全取出数据,所以之后的循环中enumerate(FileTwoReader)都一直处于空的状态就不再执行。

这时可以将第二个打开文件放到第一次循环体内,如下代码所示:

import csv
with open('tets.csv', 'r') as f:
    FileOneReader = csv.DictReader(f)
    for item, row in enumerate(FileOneReader):
        with open('test.csv', 'r') as file:
            FileTwoReader = csv.DictReader(file)
            for item2, row2 in enumerate(FileTwoReader):
                print(item, row)
                print(item2, row2)
结果为:
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
1 {'id': '2', 'name': 'Oliver', 'age': '25', 'gender': 'M'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
2 {'id': '3', 'name': 'Shayna', 'age': '18', 'gender': 'F'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
3 {'id': '4', 'name': 'Fechin', 'age': '18', 'gender': 'M'}
1 {'id': '2', 'name': 'Oliver', 'age': '25', 'gender': 'M'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
1 {'id': '2', 'name': 'Oliver', 'age': '25', 'gender': 'M'}
1 {'id': '2', 'name': 'Oliver', 'age': '25', 'gender': 'M'}
1 {'id': '2', 'name': 'Oliver', 'age': '25', 'gender': 'M'}
2 {'id': '3', 'name': 'Shayna', 'age': '18', 'gender': 'F'}
1 {'id': '2', 'name': 'Oliver', 'age': '25', 'gender': 'M'}
3 {'id': '4', 'name': 'Fechin', 'age': '18', 'gender': 'M'}
2 {'id': '3', 'name': 'Shayna', 'age': '18', 'gender': 'F'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
2 {'id': '3', 'name': 'Shayna', 'age': '18', 'gender': 'F'}
1 {'id': '2', 'name': 'Oliver', 'age': '25', 'gender': 'M'}
2 {'id': '3', 'name': 'Shayna', 'age': '18', 'gender': 'F'}
2 {'id': '3', 'name': 'Shayna', 'age': '18', 'gender': 'F'}
2 {'id': '3', 'name': 'Shayna', 'age': '18', 'gender': 'F'}
3 {'id': '4', 'name': 'Fechin', 'age': '18', 'gender': 'M'}
3 {'id': '4', 'name': 'Fechin', 'age': '18', 'gender': 'M'}
0 {'id': '1', 'name': 'Roberta', 'age': '39', 'gender': 'M'}
3 {'id': '4', 'name': 'Fechin', 'age': '18', 'gender': 'M'}
1 {'id': '2', 'name': 'Oliver', 'age': '25', 'gender': 'M'}
3 {'id': '4', 'name': 'Fechin', 'age': '18', 'gender': 'M'}
2 {'id': '3', 'name': 'Shayna', 'age': '18', 'gender': 'F'}
3 {'id': '4', 'name': 'Fechin', 'age': '18', 'gender': 'M'}
3 {'id': '4', 'name': 'Fechin', 'age': '18', 'gender': 'M'}
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是一个Python实现的读取CSV文件指定列,并保存到一个文件中的案例: 假设我们有一个名为“data.csv”的CSV文件,其中包含以下数据: ``` id,name,age,gender 1,Alice,25,Female 2,Bob,30,Male 3,Charlie,40,Male 4,Dave,35,Male 5,Emily,27,Female ``` 现在我们想要读取“name”和“gender”列,并将它们保存到一个新的CSV文件中。 ```python import csv # 源文件路径 input_file = 'data.csv' # 输出文件路径 output_file = 'output.csv' # 要读取的列 columns_to_keep = ['name', 'gender'] # 打开源文件输出文件 with open(input_file, 'r') as f_in, open(output_file, 'w', newline='') as f_out: # 使用csv模块读取csv文件 reader = csv.DictReader(f_in) # 使用csv模块写入csv文件 writer = csv.DictWriter(f_out, fieldnames=columns_to_keep) # 写入标题行 writer.writeheader() # 遍历源文件中的每一行数据 for row in reader: # 创建一个新的字典,仅包含要保留的列 new_row = {col: row[col] for col in columns_to_keep} # 将新行写入输出文件 writer.writerow(new_row) ``` 这个程序会读取“data.csv文件中的数据,并将“name”和“gender”列保存到“output.csv文件中。输出文件中的数据如下: ``` name,gender Alice,Female Bob,Male Charlie,Male Dave,Male Emily,Female ``` 注意,我们使用了`csv.DictReader`和`csv.DictWriter`来读取和写入CSV文件。这两个类允许我们使用列名来引用数据,而不是使用列索引。这使得代码更易于阅读和维护。同时,我们使用了字典推导式来创建一个新的字典,仅包含要保留的列。最后,我们使用`writer.writerow`方法将新行写入输出文件

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值