pandas 中 取一个txt第一列与另一个txt第一列交集并输出第一个txt交集后的结果
假设有两个txt文件,分别为file1.txt和file2.txt,且它们都只有一列数据。可以使用pandas库来实现取交集的操作,并输出第一个txt文件交集后的结果。
具体步骤如下:
- 使用pandas的read_csv函数读取两个txt文件,并指定header=None参数以表示文件没有列名。
import pandas as pd
df1 = pd.read_csv('file1.txt', header=None)
df2 = pd.read_csv('file2.txt', header=None)
- 使用pandas的merge函数对两个DataFrame进行合并,并指定how='inner'参数以表示取交集。
merged_df = pd.merge(df1, df2, how='inner')
- 输出交集后的结果,可以直接将DataFrame转换为list并输出。
result = merged_df.values.tolist()
print(result)
完整代码如下:
import pandas as pd
df1 = pd.read_csv('file1.txt', header=None)
df2 = pd.read_csv('file2.txt', header=None)
merged_df = pd.merge(df1, df2, how='inner')
result = merged_df.values.tolist()
print(result)
原文地址: https://www.cveoy.top/t/topic/bDUy 著作权归作者所有。请勿转载和采集!