pandas實現(xiàn)to_sql將DataFrame保存到數(shù)據(jù)庫中
目的
在數(shù)據(jù)分析時,我們有中間結(jié)果,或者最終的結(jié)果,需要保存到數(shù)據(jù)庫中;或者我們有一個中間的結(jié)果,如果放到數(shù)據(jù)庫中通過sql操作會更加的直觀,處理后再將結(jié)果讀取到DataFrame中。這兩個場景,就需要用到DataFrame的to_sql操作。
具體的操作
連接數(shù)據(jù)庫代碼
import pandas as pd
from sqlalchemy import create_engine
# default
engine = create_engine('mysql+pymysql://ledao:ledao123@localhost/pandas_learn')
original_data = pd.read_sql_table('cellfee', engine)
original_data
結(jié)果如下所示。

對數(shù)據(jù)進(jìn)行匯總,每個小區(qū)的電費進(jìn)行求和放到Series中,然后將所有小區(qū)的總電費放到DataFrame中,最后將DataFrame保存到數(shù)據(jù)庫中,代碼如下所示。
all_cells = [] for k, v in original_data.groupby(by=['cityid', 'cellid']): onecell = pd.Series(data=[k[0], k[1], v['fee'].sum()], index=['cityid', 'cellid', 'fee_sum']) all_cells.append(onecell) all_cells = pd.DataFrame(all_cells) all_cells.to_sql(name='cells_fee', con=engine, chunksize=1000, if_exists='replace', index=None)
對于DataFrame的to_sql函數(shù),需要注意的參數(shù)在代碼中已經(jīng)寫出來,其中比較重要的是chunksize、if_exists和index。
chunksize可以設(shè)置一次入庫的大??;if_exists設(shè)置如果數(shù)據(jù)庫中存在同名表怎么辦,‘replace'表示將表原來數(shù)據(jù)刪除放入當(dāng)前數(shù)據(jù);‘a(chǎn)ppend'表示追加;‘fail'則表示將拋出異常,結(jié)束操作,默認(rèn)是‘fail';index=接受boolean值,表示是否將DataFrame的index也作為表的列存儲。
最終存表的結(jié)果如下圖所示。

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
使用C#配合ArcGIS Engine進(jìn)行地理信息系統(tǒng)開發(fā)
這篇文章主要介紹了使用C#配合ArcGIS Engine進(jìn)行地理信息系統(tǒng)開發(fā),ArcGIS Engine是Windows系統(tǒng)上可以讓程序員創(chuàng)建自定義的GIS桌面程序,需要的朋友可以參考下2016-02-02
Python實現(xiàn)將圖像轉(zhuǎn)換為ASCII字符圖
使用Python進(jìn)行圖像處理,非常快捷方便,往往簡短幾行代碼就可以實現(xiàn)功能強大的效果。在這篇文章中,我們將使用Python將圖像轉(zhuǎn)換為ASCII字符照,感興趣的可以了解一下2022-08-08
Django中select_related和prefetch_related的用法與區(qū)別詳解
在實際的開發(fā)中,模型之間經(jīng)常存在復(fù)雜的關(guān)聯(lián)關(guān)系,下面這篇文章主要給大家介紹了關(guān)于Django中select_related和prefetch_related的用法與區(qū)別的相關(guān)資料,需要的朋友可以參考下2022-11-11

