Python 比较文件 A 和 B 中的值的方法，如果 A 匹配，则打印 B 中的整行

Question

我有两个文本文件:

我的文件1:

sports1
sports2
sports3
sports4
sports5
sports6
sports7
sports8

我的文件2:

sports1  Cricket
sports2  golf
sports3  Hocky
sports4
sports5  Chess
sports6  Snooker
sports7  Foosball
sports8  Tampts

从上面的两个文件中，f myfile1 中的第一列与 myfile2 中的第一列匹配，那么它应该打印 myfile2 的两列。

下面的 awk 单行代码可以工作，但我正在 Python 中寻找类似的代码。

awk 'NR==FNR{c[$1]++;next};c[$1]' myfile1 myfile2

Answer 1

如果您愿意，您可以使用第三方库，例如 Pandas。这可能是最接近你的一句台词了。

delim_whitespace=True 确保 Pandas 使用空格来分隔列，而不是常规 csv 中所期望的 ,。

import pandas as pd

df1 = pd.read_csv('file1.txt', header=None, names=['sport'])
df2 = pd.read_csv('file2.txt', header=None, names=['sport', 'name'],
                  delim_whitespace=True)

res = df2[df2['sport'].isin(df1['sport'].unique())]

print(res)