根据另一个 csv 文件过滤 csv 文件中的行并将过滤后的数据保存在新文件中

大家好

所以我试图根据file1过滤file2，其中file1是file2的子集。但 file2 有一个描述列，我需要能够分析 file1 中的数据。我想做的是过滤 file2，并仅获取 file1 中的标题及其描述。我尝试了这个，但我不太确定它是否完全正确，而且它可以工作，但我的计算机上没有保存任何文件。

import re
import mmap
from pandas import DataFrame
output = []
with open('file2.csv', 'r') as f2:
    mm = mmap.mmap(f2.fileno(), 0, access=mmap.ACCESS_READ)
    for line in open('file1.csv', 'r'):
        Title = bytes("")
        nameMatch = re.search(Title, mm)
        if nameMatch:
            # output.append(str(""))
            fulltypes = [ 'O*NET-SOC Code', 'Title' , 'Discription' ]
            final = DataFrame(columns=fulltypes)
            final.to_csv(output.append(str("")))
    mm.close()

任何想法？

假设您的 csv 文件不太大，您可以通过将两者读入来完成此操作pandas并使用join方法。举个例子：

import pandas as pd

file1 = pd.DataFrame({'Title': ['file1.csv', 'file2.csv', 'file3.csv']})
file2 = pd.DataFrame({'Title': ['file1.csv', 'file2.csv', 'file4.csv'],
                      'Description': ['List of files', 'List of descriptions', 'Something unrelated']})

joined = pd.merge(file1, file2, left_on='Title', right_on='Title')

print joined

这打印：

       Title           Description
0  file1.csv         List of files
1  file2.csv  List of descriptions

即仅存在于两者中的文件。

As pandas可以本机将 csv 读入数据帧，在您的情况下您可以这样做：

import pandas as pd

file1 = pd.DataFrame.from_csv('file1.csv')
file2 = pd.DataFrame.from_csv('file2.csv')

joined = pd.merge(file1, file2, left_on='Title', right_on='Title')

joined.to_csv('Output.csv', index=False)