1. 程式人生 > 實用技巧 >python資料庫操作

python資料庫操作

一、使用的庫

1.連線資料庫pysql
2.讀取資料庫pandas
3.將讀取的元組資料型別轉換為列表numpy

二、連線資料庫

import pymysql.cursors
db = pymysql.connect(
        host='**',
        port=**,
        user='**',
        password='**',
        db='**',

        charset='utf8'
        )
# 建立一個遊標
cursor = db.cursor()
#執行sql
cursor.execute(sql)

三、讀取資料庫返回結果

1. 讀取一行或多行資料

  • fetchone()獲取結果集的下一行
  • fetchall()獲取結果集中的剩下的所有行——<class 'tuple'>

2.讀取一列資料

  • python讀取sql裡面的指定資料列(一列資料),並將其轉換成列表使用
    • 用pandas來讀取資料庫裡面的內容——<class 'pandas.core.frame.DataFrame'>
    • 再使用numpy庫將DataFrame資料轉換成陣列——<class 'numpy.ndarray'>
    • 再將陣列型別資料轉換成列表——<class 'list'>
# 讀取資料
import pandas as pd
import numpy as np

result = cursor.fetchall()
print(result,type(result))
result1 = pd.DataFrame(list(result), columns=["sku_count", "sku_pay_amount", "unit_price"])
print(result1,type(result1))
# 先使用array()將DataFrame轉換一下
result2 = np.array(result1)
print(result2,type(result2))
# 再將轉換後的資料用tolist()轉成列表
result3 = result2.tolist()
print(result3,type(result3))
for i in range(0, len(result3)):
    value_column = result3[i][0]
    print(value_column)