import pandas as pd
import os
img_expand = pd.read_csv('img_expand.csv')
img_test = pd.read_csv('test_list.csv')
print(img_test.shape)
print(img_expand.shape)
img_expand_train=img_expand[~img_expand['filename'].isin(img_test['filename'])]
print(img_expand_train.shape)
结果:
(3061, 2)
(103882, 2)
(100821, 2)