使用 python/pymongo 将 json 数据写入 csv 文件

Question

我正在从 mongodb 获取数据并使用 csv.DictWriter 方法将 json 值写入 csv 文件，但是在写入时，我在每一行之后都得到一个空行。

如何避免这些空行？

代码片段：（我正在做一些我在这里没有包括的数据操作）

with open('test_output.csv', 'w') as csvfile:
  fields = ['date', 'ns', 'storageSize']
  writer = csv.DictWriter(csvfile, fieldnames=fields)
  writer.writeheader()
  stats = client['db_stats'].coll_stats.find({})
  for x in stats:
     writer.writerow({'date': x["date"], 'ns': x["ns"], 'storageSize': x["storageSize"]});

输出：（每次写入后有空行）

Answer 1

您可以使用 pandas 事后解决问题：

import pandas as pd # import pandas
data = pd.read_csv('test_output.csv') # pull in csv file
data = data.iloc[::2] # chop every 2nd row from frame
data.to_csv('test_output.csv',index=False) # save over csv

您应该可以将上面的 sn-p 添加到现有代码的末尾。