将 API 的响应保存到 csv 文件

Question

我是 python 的新手。我正在使用一些调用 api 并以 csv 格式获得响应的 python 代码。我想知道如何将 csv 响应保存到 csv fie。

#! /usr/bin/env python
import httplib2
# These aren't needed, just for this example
from pprint import pformat

from datetime import datetime
import pytz

from pvlive_api import PVLive

import pandas as pd
import json


def post_elexon(url):
    http_obj = httplib2.Http()
    resp, content = http_obj.request(
        uri=url,
        method='GET',
        headers={'Content-Type': 'application/xml; charset=UTF-8'},)
    return resp, content

def main():
    resp, content = post_elexon(url='https://api.bmreports.com/BMRS/B1770/v1?APIKey=MY_API_KEY&SettlementDate=2015-03-01&Period=1&ServiceType=csv',)
    print ("===Response===")
    print (resp)
    
    print ("===Content===")
    print (pformat(content))
    print ("===Finished===")


if __name__ == "__main__":
    main()

任何帮助，建议将不胜感激。

谢谢

Answer 1

试试这个：

    import csv
    
    with open('out.csv', 'w') as f:
        writer = csv.writer(resp)
        for line in resp.iter_lines():
            writer.writerow(line.decode('utf-8').split(','))

编辑：

我测试了您的请求 - 它返回一个 json。

所以你可以把它保存为json：

with open('response.json', 'w') as f:
    json.dump(resp, f)