(資料科學學習手札51)用pymysql來操控MySQL資料庫
阿新 • • 發佈:2018-12-17
from sklearn.datasets import load_iris '''獲取鳶尾花資料,其中X,y為二維陣列''' X,y = load_iris(return_X_y=True) '''建立指定鳶尾花類別名稱的列表''' Species = ['setosa','versicolor','virginica'] '''通過操作遊標執行SQL語句,以建立iris表''' cur.execute("CREATE TABLE IF NOT EXISTS iris" "(Sepal_Length float," "Sepal_Width float," "Petal_Length float," "Petal_Width float,Species char(20)" ")") '''構造將X,y資料一次性插入iris的SQL語句''' BaseSQL = "INSERT INTO iris VALUES" for i in range(X.shape[0]): BaseSQL += "({},{},{},{},'{}'),".format(X[i,0],X[i,1],X[i,2],X[i,3],Species[y[i]]) '''去除末尾多餘的逗號''' BaseSQL= BaseSQL[:-1] '''執行插入鳶尾花資料的SQL語句''' cur.execute(BaseSQL)