python - 如何使用 Python 中的 API 从 csv 文件读取数据

Question

我正在尝试从以下链接下载数据:
NHTSA Website

如果我创建如下链接，它将在我的计算机上下载 CSV 文件 http://webapi.nhtsa.gov/api/SafetyRatings/modelyear/2019/make/ACURA/model/RDX?format=csv

如何使用 Python 中的 API 读取不同车辆的此文件？

这是迄今为止我的代码:

db = sql.connect("localhost","root","password","TEST")

cursor = db.cursor()

sql = "SELECT * FROM TEST.car_model"
cursor.execute(sql)

data = cursor.fetchall()

for row in data:
    apiUrl = "http://webapi.nhtsa.gov/api/SafetyRatings/modelyear/"
    apiParams = str(row[1])+"/make/"+row[2].replace(" ","%20")+"/model/"+row[3].rstrip().replace(" ","%20")
    apiFormat = "?format=csv"
    link = apiUrl + apiParams + apiFormat
    response = urlopen(apiUrl + apiParams + apiFormat)

    f = open(link, 'rb')
    reader = csv.reader(f)
    for row in reader:
        print(row)

db.close()

response 会自动将文件下载到我的“下载”文件夹中吗？

Answer 1

你可以试试这个

model = input("Enter model name: ")
year = input("year: ")

url = "http://webapi.nhtsa.gov/api/SafetyRatings/modelyear/"+year+"/make/ACURA/model/"+model+"?format=csv"

import urllib.request

with urllib.request.urlopen(url) as response:
    html = response.read()

with open(model+year+".csv", "w") as f:
    f.write(html)

您可以向您的网址添加更多变量。然后，您可以使用 pandas 包进行阅读。

import pandas as pd
vechiles = pd.read_csv("vichles.csv")