使用 python 3 代码在 CSV 文件中查找特定标头

现在我有Python 3代码，它获取CSV文件中的一列数据，根据空格将每个单元格中的短语分隔成单独的单词，然后将数据导出回新的CSV文件中。

我想知道是否有一种方法可以告诉 python 仅将格式化代码应用于具有特定标题的特定列？

这是我的源数据的样子

Keyword              Source       Number 
Lions Tigers Bears     US          3
Dogs Zebra            Canada       5
Sharks Guppies         US          2

这是我的代码，它将每个单元格中的短语根据空格分隔成单独的单词

with open(b'C:\Users\jk\Desktop\helloworld.csv', 'r') as datafile:
    data = []
    for row in datafile:
        data.extend(item.strip() for item in row.split())
with open('test.csv', 'w') as a_file:
    for result in data:
        result = ''.join(result)
        a_file.write(result + '\n')
        print(result)

这样源数据就变成了

 Keywords         Source         Number
 Lions            US              3
 Tigers
 Bears
 Dogs             Canada          5

ETC

在这种情况下，我只需要将所有这些代码应用于具有标题的一列Keyword。理想情况下，我想做的是将“源”和“数字”中找到的数据扩展到这些新创建的行（Lions US 3 - Tigers US 3 - Bears US 3 等），但我还没有真正做到还没弄清楚那部分！

我在论坛上闲逛了一段时间，试图找到答案，我知道你可以告诉 python 读取 CSV 文件的第一行，其中放置了标题（headers = file.readline()）但除此之外我就迷失了。使用 CSV 阅读器可以更轻松地完成此任务吗？

Use the csv module https://docs.python.org/3/library/csv.html将数据拆分为列。使用csv.DictReader()对象，以便更轻松地通过标题选择列：

import csv

source = r'C:\Users\jk\Desktop\helloworld.csv'
dest = 'test.csv'

with open(source, newline='') as inf, open(dest, 'w', newline='') as outf:
    reader = csv.DictReader(inf)
    writer = csv.DictWriter(outf, fieldnames=reader.fieldnames)
    for row in reader:
        words = row['Keyword'].split()
        row['Keyword'] = words[0]
        writer.writerow(row)
        writer.writerows({'Keyword': w} for w in words[1:])

The DictReader()将从文件中读取第一行并将其用作为每行生成的字典的键；所以一行看起来像：

{'Keyword': 'Lions Tigers Bears', 'Source': 'US', 'Number': '3'}

现在，您可以单独处理每一列，并仅使用该列的第一个单词更新字典Keyword为剩余单词生成附加行之前的列。

我假设你的文件是comma分开了。如果需要不同的分隔符，则设置delimiter该角色的参数：

reader = csv.DictReader(inf, delimiter='\t')

用于制表符分隔的格式。有关各种选项，请参阅模块文档，包括称为dialects.

Demo:

>>> import sys
>>> import csv
>>> from io import StringIO
>>> sample = StringIO('''\
... Keyword,Source,Number
... Lions Tigers Bears,US,3
... Dogs Zebra,Canada,5
... Sharks Guppies,US,2
... ''')
>>> output = StringIO()
>>> reader = csv.DictReader(sample)
>>> writer = csv.DictWriter(output, fieldnames=reader.fieldnames)
>>> for row in reader:
...     words = row['Keyword'].split()
...     row['Keyword'] = words[0]
...     writer.writerow(row)
...     writer.writerows({'Keyword': w} for w in words[1:])
... 
12
15
13
>>> print(output.getvalue())
Lions,US,3
Tigers,,
Bears,,
Dogs,Canada,5
Zebras,,
Sharks,US,2
Guppies,,