如何在多个csv中搜索值并在python中创建新的csv结果？

Question

I have 5 CSV files with multiple columns and one more csv file with list of ID's. i have to search ID's across all 5 CSV files and create a new CSV file with results like if value present mark 1 if not mark 0.

for example :

1.csv

ID| NAME |ORG
1 | A |B
2 | C |D
7 | X |Y

2.csv

ID |NAME |ORG
3 |E |F
4 |G |H
7 |V |U

id.csv

ID
1
2
3
4
7

now result.csv should look like.

result.csv

ID | 1.csv |2.csv
1 | 1 |0
2 | 1 |0
3 | 0 |1
4 | 0 |1
7 | 1 |1

Answer 1

here is a simple implementation with pandas and glob .

import glob
import pandas as pd

csv_files = glob.glob('[0-9].csv')
id_file = pd.read_csv('id.csv', sep='|', index_col=0)
res = pd.DataFrame(0, index=id_file.index, columns=csv_files)

for f in csv_files:
    tmp = pd.read_csv(f, sep='|', index_col=0)
    res.loc[id_file.index.isin(tmp.index), f] = 1

res.to_csv('result.csv', sep='|')