將數據附加到 CSV 文件並刪除列表括號？

Question

我構建了一個網絡抓取類，現在正嘗試使用這種方法將平均成本和其他值寫入 csv 文件——

def avg_data_to_csv(self):
    avg_data = self.avg_data
    df = pd.DataFrame(avg_data, columns=['Location', 'Average Price', 'Average Price per SQFT'])
    csv = df.to_csv('AVG_DATA.csv', index=False)
    return csv

然后我編寫了另一種方法，每次運行程序時將更新的平均值添加到新行中的同一個 csv 文件中——

def add_to_avg_csv(self):
    new_data = self.avg_data
    with open('AVG_DATA.csv', 'a') as obj:
        writer_object = csv.DictWriter(obj, fieldnames=self.avg_data)
        writer_object.writerow(new_data)
        obj.close()

但是，當我對此進行測試時，我的 csv 文件如下所示 -

兩次運行后的 csv 文件

我無法弄清楚如何在沒有列表括號的情況下添加數據。 當我將新數據添加到文件時，我將如何刪除列表括號？

Answer 1

由於您使用的是 writerow() 最簡單的方法是讓函數 avg_data_to_csv() 返回新數據記錄的字符串而不是單行數據幀。

另一種方法是用 pandas 打開“AVG_DATA.csv”，連接/附加 avg_data_to_csv() 的輸出，然后將其保存為“AVG_DATA.csv”。