python 如何删除两个文件夹中不同的文件

例.

文件夹A内有 1.2.4.
文件夹B内有 1.2.3.
要求把文件名相同的留下 把文件名不同的删掉

# 思路：
# 1、分别获取两个文件夹的文件，并相应转化为集合
# 2、根据差集删除相应文件
import os
 
set_A = set(os.listdir(path_A))
set_B = set(os.listdir(path_B))
 
for i in set_A - set_B: os.remove(os.path.join(path_A, i))
for i in set_B - set_A: os.remove(os.path.join(path_B, i))

 后缀不同呢 A是.xml B是.jpg怎么处理？

import os
 
set_A = set(i.split('.')[0] for i in os.listdir(path_A))
set_B = set(i.split('.')[0] for i in os.listdir(path_B))
 
for i in set_A - set_B: os.remove(os.path.join(path_A, i + '.xml'))
for i in set_B - set_A: os.remove(os.path.join(path_B, i + '.jpg'))

方法二，不推荐：

 
import os
 
path1 = r'D:\test\test1\\'
path2 = r'D:\test\test2\\'
 
filenames1 = os.listdir(path1)
filenames2 = os.listdir(path2)
 
samefiles = [x for x in filenames1 if x in filenames2] #两个文件夹表都存在
 
dfiles1 = [y for y in filenames1 if y not in samefiles] #两个文件夹中的不同元素
dfiles2 = [y for y in filenames2 if y not in samefiles] #两个文件夹中的不同元素
 
os.remove(path1 + dfiles1[0])
os.remove(path2 + dfiles2[0])