将数据写入 CSV 格式文件

Question

我正在尝试使用从 MongoDB 中获取的数据和 Mongopy 将一些数据写入 CSV 格式。当前，当前设置了 headers，问题是实际数据没有被插入到 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.DictWriter 而不是 csv.writer 因为您的查询结果是字典。您还需要在查询中指定 projection 字段。将您的查询更改为此。

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)

将数据写入 CSV 格式文件

Writing Data Into CSV Format File

python

mongodb

csv

pymongo

export-to-csv