How to find all elements of the same type with python