Advertisement
Guest User

Untitled

a guest
Jun 30th, 2015
132
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.60 KB | None | 0 0
  1. from types import GeneratorType
  2.  
  3.  
  4. def remove_duplicates(iterable, key=None):
  5. iterable_class = type(iterable)
  6. if iterable_class in [dict, set]:
  7. raise TypeError("You can't remove duplicates from a {}!".format(iterable_class))
  8.  
  9. if not key:
  10. _key = lambda a: a
  11. elif not callable(key):
  12. _key = lambda a: a[key]
  13. else:
  14. _key = key
  15.  
  16. s = set()
  17. gen = (s.add(_key(i)) or i for i in iterable if _key(i) not in s)
  18.  
  19. if iterable_class == str:
  20. return ''.join(gen)
  21. elif iterable_class == GeneratorType:
  22. return list(gen)
  23.  
  24. return iterable_class(gen)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement