如何在 python(pandas) 中保存數據

Question

當我將數據保存到 csv 文件時，只會保存 ID 為 103 的記錄，而 ID 為 101 和 102 的記錄不可用。

import pandas as pd
import requests as rq

for vehicleList in range(101, 103):
    vehicleList = vehicleList + 1
    x = str(vehicleList)

    r = rq.get('https://api.tfl.gov.uk/vehicle/' + x + '/arrivals')
    r = r.text
    df = pd.read_json(r)

    df.to_csv('filename.csv')

Answer 1

您需要在保存之前連接數據。 例如：

import pandas as pd
import requests as rq

temp = pd.DataFrame()
df = pd.DataFrame()

for vehicleList in range(101, 103):
    vehicleList = vehicleList + 1
    x = str(vehicleList)
    r = rq.get('https://api.tfl.gov.uk/vehicle/' + x + '/arrivals')
    r = r.text
    temp = pd.read_json(r)

    # some identifier here  
    temp['Type'] = x

    # concat
    df = pd.concat([df, temp]).reset_index(drop=True)

df.to_csv('filename.csv')

您將每個循環temp分配給一個臨時 DataFrame，然后添加一個唯一的 id 來表示組temp['Type'] ，然后連接到df 。 循環完成后，您將保存包含所有迭代的 df。