如何使用 csv 阅读器将列从一个文件映射到另一个文件

Question

我是 Python csv.reader 的新手，我正在尝试使用 sheet1 中的标题填充 sheet2 中的数据：

sheet1.csv 包含以下标题

  1         2         3         4         5         6         7  

sheet2.csv 包含以下数据

  col_1     col_4     col_5     col_7
 yellow      Blue     Green       Red

想要的输出如下：

      1         2         3         4         5         6         7  
 Yellow                          Blue     Green                 Red

最初我使用pandas 完成此操作，但我想在python 中使用csv.reader 完成此操作。

这是我目前的尝试：

import csv

with open('sheet1.csv', newline='') as myFile:
    reader = csv.reader(myFile)
    for row in reader:
        print(row)

Answer 1

我知道我以前回答过这个问题，但我懒得去找原帖。

[mre] 版本：

from io import StringIO
import csv
import sys

# map column names to template
mapper = lambda fieldname: fieldname.split('_')[-1]

template = '''\
1,2,3,4,5,6,7
'''

source = '''\
col_1,col_4,col_5,col_7
Yellow,Blue,Green,Red
White,Black,Cyan,Magenta
'''

fieldnames = csv.DictReader(StringIO(template)).fieldnames

out = csv.DictWriter(sys.stdout, fieldnames)
out.writeheader()

reader = csv.DictReader(StringIO(source))
for row in reader:
    out.writerow({mapper(k): v for (k, v) in row.items()})

带有实际文件版本：

import csv

# map column names to template
mapper = lambda fieldname: fieldname.split('_')[-1]

with open('template.csv', newline='') as template:
    fieldnames = csv.DictReader(template).fieldnames

with open('source.csv', newline='') as source, open('output.csv', 'w', newline='') as output:
    reader = csv.DictReader(source)

    out = csv.DictWriter(output, fieldnames)
    out.writeheader()

    for row in reader:
        out.writerow({mapper(k): v for (k, v) in row.items()})

输出：

1,2,3,4,5,6,7
Yellow,,,Blue,Green,,Red
White,,,Black,Cyan,,Magenta