将数据写入CSV格式文件

Question

我正在尝试使用从MongoDB获取的数据编写一些数据到CSV格式。目前，标头是当前设置的，问题是实际数据没有插入到CSV文件中。下面是代码片段：

from pymongo import MongoClient
import csv
import os

conn = pymongo.MongoClient()

db = conn.clixster_dev

cursor = db.channels.find({},{'_id':0 , 'company-reg-no':0, 'isdel':0 , 'last_off':0 , 'last_on':0 , 'online':0 , 'password':0 , 'psotcode':0 , 'state':0 , 'subagent':0})


outfile = open( "asdxk.csv", "w" )

# get a csv writer
 writer = csv.writer( outfile )

# write header
 writer.writerow(['postcode', 'upline', 'cit_state', 'contact_person', 'contact_no','nominal_level', 'credit', 'level', 'account_holder', 'merchantid', 'email', 'bank', 'reg_date' ,'address','acc_no','company_name'])

# write data
[ writer.writerow(x) for x in cursor ]

# close file
outfile.close()

任何考虑都是感激的，

Answer 1

您需要使用csv.writer而不是csv.DictWriter，因为查询结果是字典。还需要在查询中指定投影字段。将查询更改为以下内容。

cursor = db.channels.find({},{'_id':0, 'postcode': 1, 'upline': 1, 'cit_state': 1, 'contact_person': 1, 'contact_no': 1,'nominal_level': 1, 'credit': 1, 'level': 1, 'account_holder': 1, 'merchantid': 1, 'email': 1, 'bank': 1, 'reg_date': 1 ,'address': 1,'acc_no': 1,'company_name': 1})

并使用with语句

with open('asdxk.csv', 'w') as outfile:
    fields = ['postcode', 'upline', 'cit_state', 'contact_person', 'contact_no','nominal_level', 'credit', 'level', 'account_holder', 'merchantid', 'email', 'bank', 'reg_date' ,'address','acc_no','company_name']
    writer = csv.DictWriter(outfile, fieldnames=fields)
    writer.writeheader()
    for x in cursor:
        writer.writerow(x)