固定sql語句傳參批量查詢數據庫腳本

#!/usr/bin/env python                                                                               
# encoding: utf-8                                                                   
                                                                                    
import pandas as pd                                                                 
import psycopg2                                                                     
conn_dv = psycopg2.connect(database="", user="", password="")
                                                                                    
def read_csv(filename):                                                                     
    usr_list = pd.read_csv(filename)                                       
    res  = []                                                                       
    for i in usr_list.values:                                                       
        res.extend(i)                                                               
    return tuple(res)                                                               
                                                                                    
                                                                                    
def sql_to_csv(sql, conn):                                                          
    df = pd.read_sql_query(sql, conn)                                               
    df.to_csv('tmp.csv', index=False)                                               
    return df                                                                       
                      
                                                                                    
if __name__=='__main__':                                                            
    usr_ids = read_csv('name.csv')                                                            
    sql = """select * from usr_info where cust_id = {uid};""".format(uid=usr_ids)
    df = sql_to_csv(sql, conn_dv)    
                                               

關鍵知識點回顧：

• pd.read_csv() 讀取csv
• pd.read_sql_query(sql,conn) 執行sql查詢方法,sql為語句，conn為數據庫連接字符串
• df.to_csv('tmp.csv', index=False) 去除index對象，生成csv的名字為tmp.csv