首頁>科技>

本文1542字,建議閱讀4分鐘。

本文以17個 txt 文字,3萬多條資料為例,使用 Python 連線 MySQL 資料庫,實現快速操作。

應用場景:工作中經常遇到大量的資料需要整合、去重、按照特定格式匯出等情況。如果用 Excel 操作,不僅費時費力,還不準確,有麼有更高效的解決方案呢?

本文以17個 txt 文字,3萬多條資料為例,使用 Python 連線 MySQL 資料庫,實現快速操作。

別人加班乾的活,我的 Python 小助手幾秒鐘就搞定了!

本文主要包括以下三方面內容:

資料寫入資料去重資料匯出

將資料寫入 MySQL 資料庫

下圖所示檔案是本文的資料來源:

我們的設想是:編寫一個小程式,扔在這個資料夾裡,雙擊後就可以自動讀取每個 txt 文件中的資料,並寫入資料庫。

重點程式碼解釋:

這段程式碼用到了 pymysql 和 os 兩個庫。

pymysql:用來操作 MySQL 資料庫;os:用來遍歷所在資料夾下的所有檔案。

現將主要程式碼解釋如下:

1、遍歷任意資料夾下所有檔名稱

程式寫好後,使用 pyinstaller 打包成 exe 程式,並放在要操作的資料夾下面。

通過 path = os.getcwd()命令,獲取該 exe 檔案所在目錄。

通過 files = os.listdir(path)命令,獲取 exe 檔案所在目錄下的所有檔名稱,並存入 files 列表中。

這樣我們就獲得了所有的 txt 檔名稱,你可以任意命名你的 txt 檔名,程式都能讀出來。

2、將資料寫入資料庫

(1)連線資料庫,並在資料庫中建立新表

A. 連線到我的 qq 資料庫

conn = pymysql.connect(host='localhost', user='root', password='123456', db='qq', charset='utf8')

B. 建立新表 qq

在 qq 資料庫中建立新表,表名稱為 qq ,包含 2 個欄位:id 欄位為主鍵、自動遞增;qq 欄位為字元型,用於儲存資料。

cur.execute("CREATE TABLE qq ( id int(5) NOT NULL auto_increment, qq varchar(20)NOT NULL, PRIMARY KEY (id))")

(2)將資料寫入資料庫

這裡使用了兩層迴圈:

for file in files:    f = open(file,'r',encoding = 'UTF-8')    next(f)    for line in f:        i += 1        #print(line)        sql = "insert into qq(qq) values(%s);"        cur.execute(sql,line)        print("插入第", i, "條資料!")        conn.commit()    f.close()

第一層迴圈是用來依次開啟上述 17 個 txt 檔案。

第二層迴圈是依次讀取每個 txt 檔案的每一行,並將改行資料插入資料庫表 qq 的 qq欄位。

至此就完成了資料的匯入,總共32073條資料。

資料清洗

這裡以去除重複值為例,簡單介紹一下資料清洗。

1、建立一個新表,用來儲存清洗後的資料

可以在 cmd 視窗下登陸 MySQL,開啟 qq 資料庫,執行下列操作:

CREATE TABLE qq_dist ( id int(5) NOT NULL auto_increment, qq varchar(20)NOT NULL, PRIMARY KEY (id));

這樣就建立了新表 qq_dist ,用來儲存清洗後的資料,便於後期呼叫。

2、清洗資料

登陸 MySQL 後,執行下列操作:

insert into qq_dis(qq) select distinct qq from qq;

將從 qq 表中查找出的不重複的 qq 欄位內容,插入到 qq_dist 表中的 qq 欄位。

將資料按照特定格式匯出

案例:將清洗後的資料的第101-200行匯出到新的 txt 文字中。

程式碼如下:
import pymysqlconn = pymysql.connect(host='localhost', user='root', password='123456', db='wxid', charset='utf8')print("寫入中,請等待……")cur = conn.cursor()sql = "select wxid from wd_dis limit 100,100;"cur.execute(sql)conn.commit()alldata = cur.fetchall()f = open('data101-200.txt','a')i = 0for data in alldata:    i += 1    f.write(data[0])    f.flush()f.closecur.close()conn.close()print("寫入完成,共寫入{}條資料!".format(i))
重點程式碼解釋:

1、 limit

MySQL 中 limit m,n 函式的含義是:從第 m+1 行開始讀取 n 行。

所以,本案例中讀取第101-200行,就是 limit 100,100

2、flush()

flush()函式一定要加上,它可以將緩衝區的資料寫入檔案中。否則就會出現生成的 txt 文件為空白的錯誤。

—完—

最新評論
  • 整治雙十一購物亂象,國家再次出手!該跟這些套路說再見了
  • 跨境電商平臺Jumia聯合創始人分析非洲機遇 留給你的蛋糕有多大?