import os

l=os.listdir("/home/user")
py_files = filter (lambda name:name.endswith(".py"),l)
print py_files