Extract extension from filename in Python

Use pathlib's Path object to find extension with suffix property. Slice away the period if you don't need it.