Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/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)
Add Comment
Please, Sign In to add comment