csvwrite.py 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. #coding=utf-8
  2. '''
  3. Created on 2016年7月29日
  4. @author: uas
  5. '''
  6. from util_common import Constant
  7. import csv
  8. from pymongo.mongo_client import MongoClient
  9. cli=MongoClient(Constant.MONGODB_URL)
  10. db=cli.spider
  11. result=db.panel_propertyvalue_0728.find()
  12. total_list=list()
  13. for index, r in enumerate(result):
  14. print(index)
  15. row_list=list()
  16. row_list.append(r['kind'])
  17. row_list.append(r['property'])
  18. row_list.append(r['detno'][-4:])
  19. if row_list not in total_list:
  20. total_list.append(row_list)
  21. with open('csv_test.csv','w',newline='')as csvfile:
  22. writer=csv.writer(csvfile)
  23. writer.writerow(['器件类目','器件属性','属性所属'])
  24. for index,r in enumerate(total_list):
  25. writer.writerow(r)
  26. # csvfile =open('csv_test.csv', 'wb')
  27. # writer = csv.writer(csvfile)
  28. # csv_out.writerow(['Domain:','Mail Server:','TLS:','# of Employees:','Verified:'])
  29. #
  30. # data = [
  31. # ('小河', '25', '1234567'),
  32. # ('小芳', '18', '789456')
  33. # ]
  34. # writer.writerows(data)
  35. #
  36. # csvfile.close()