Advertisement
Guest User

Untitled

a guest
Oct 20th, 2019
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.63 KB | None | 0 0
  1. # -*- coding: utf-8 -*-
  2. num=0
  3. file_name = 'origin'
  4. file_ext = '.csv'
  5.  
  6. first_line = True
  7. files = {}
  8. file_idx = 0
  9.  
  10. with open(file_name+file_ext, 'r') as f:
  11. for line in f.readlines():
  12. if first_line == True:
  13. first_line = line
  14. continue
  15. key=line.split(",")[num]
  16. if key not in files:
  17. file_idx+=1
  18. target = open(file_name + str(file_idx) + file_ext,'w')
  19. target.write(first_line)
  20. target.write(line)
  21. files[key] = target
  22. else:
  23. target = files.get(key)
  24. target.write(line)
  25. for f in files:
  26. files.get(f).close()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement