python – 在两个文件夹中查找名称的相同初始部分的文件

发布时间：2020-12-20 11:03:41 所属栏目：Python 来源：网络整理

导读：我用listdir读取两个文件夹中的文件： from os import listdir list_1 = [file for file in listdir("./folder1/") if file.endswith(".csv")]list_2 = [file for file in listdir("./folder2/") if file.endswith(".json")] 现在我有两个清单： list_1 = ['

我用listdir读取两个文件夹中的文件：

from os import listdir 
list_1 = [file for file in listdir("./folder1/") if file.endswith(".csv")]
list_2 = [file for file in listdir("./folder2/") if file.endswith(".json")]

现在我有两个清单：

list_1 = ['12_a1_pp.csv','32_a3_pp.csv','45_a17_pp.csv','81_a123_pp.csv']
list_2 = ['12_a1.json','32_a3.json','61_a54.json']

我想找到相应的两个子列表,其中包含名称的初始部分相同的那些文件.换一种说法：

list_1b = ['12_a1_pp.csv','32_a3_pp.csv']
list_2b = ['12_a1.json','32_a3.json']

我怎样才能做到这一点？

PS注意,listdir部分可能无关紧要回答问题.我只包括它,因为如果listdir的结果保证按字母顺序排列,那么这可能有助于遍历这两个列表.当然,在这个简单的例子中,列表很短,但在实际使用案例中,它们汇集了数百个文件.

解决方法

更加pythonic的方法将使用&集合的(交集)运算符：

common = set(x[:-7] for x in list_1) & set(x[:5] for x in list_2)
list_1b = [x + '_pp.csv' for x in common]
list_2b = [x + '.json' for x in common]

编辑：如果你需要拆分每个列表的特定字符(请参阅注释),这里是一个更新版本(搜索list_1中的最后一个’_’并搜索list_2中的最后一个’.’)：

common = set(x[:x.rindex('_')] for x in list_1) & set(x[:x.rindex('.')] for x in list_2)

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!