Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def sql_query(query):
- conn = pymysql.connect(host="hostname, user="user1", password="pass",
- database="dab1", port=3306)
- cursor = conn.cursor()
- cursor.execute(query)
- rows = cursor.fetchall()
- fields = []
- for field in cursor.description:
- fields.append(field[0])
- fields = tuple(fields)
- data = [i for i in rows]
- df = pd.DataFrame.from_records(data, columns=fields)
- cursor.close()
- conn.close()
- return df
Add Comment
Please, Sign In to add comment