pandas 中 取一个txt第一列与另一个txt第一列的交集并输出第一个的表格
假设第一个txt文件名为 file1.txt,第二个txt文件名为 file2.txt,可以按照以下步骤进行操作:
- 使用 pandas 的 read_csv 函数读取两个txt文件,并指定分隔符为 '\t',同时只读取第一列:
import pandas as pd
df1 = pd.read_csv('file1.txt', sep='\t', usecols=[0], header=None)
df2 = pd.read_csv('file2.txt', sep='\t', usecols=[0], header=None)
- 使用 pandas 的 merge 函数将两个数据框按照第一列进行内连接:
merged_df = pd.merge(df1, df2, on=0, how='inner')
- 输出第一个数据框:
print(df1)
完整代码如下:
import pandas as pd
df1 = pd.read_csv('file1.txt', sep='\t', usecols=[0], header=None)
df2 = pd.read_csv('file2.txt', sep='\t', usecols=[0], header=None)
merged_df = pd.merge(df1, df2, on=0, how='inner')
print(df1)
原文地址: https://www.cveoy.top/t/topic/bDTV 著作权归作者所有。请勿转载和采集!