python - Compare multiple file name with the prefix of name in same directory

Question

I have multiple .png and .json file in same directory . And I want to check where the files available in the directory are of same name or not like a.png & a.json, b.png & b.json

Answer 1

You may try this:

import os

_, _ ,files = os.walk('.').next()
json = [f[:-5] for f in files if f.endswith('.json')]
png  = [f[:-4] for f in files if f.endswith('.png')]

json_only = set(json) - set(png)
png_only = set(png) - set(json)

json_and_png = set(json) & set(png)

... etc...