python - 查找特定词汇并提取 csv 行并在新 csv 文件中进行转换

Question

我有 30,000 行 csv 文件 列表，我想在找到特定单词后搜索这些行和列，将该行写入新的 csv 文件中。

例如:

job ,position , email , adress 

joe  ,validi ,  ceo  ,  joevalidi@yahoo.com

我想知道谁是 ceo，然后将其提取并将其保存在不同的 csv 文件中。

Answer 1

假设这是您的输入文件test.txt，位于/Users/redwing/temp/

/Users/cat/meow/meoew
/Users/cat/meow/meoew2
/Users/dog/bow/wow

然后此脚本将输出一个仅包含包含 cat 的行的文件。

in_fname = '/Users/redwing/temp/test.txt'
out_fname = '/Users/redwing/temp/test_output.txt'

f_out = open(out_fname, 'w+')
f_in = open(in_fname, 'r')
for content in f_in:
    if 'cat' in content:
        f_out.write(content)
f_in.close()
f_out.close()

此解决方案的优点是所有文件都不会读入内存，因此过滤 3000 或 3000 万行并不重要。

PS。在您编辑之前提交了我的回复。显然，您只需将 cat 替换为 ceo )