怎么把sql查询结果插入python数组

Question

如题所述

Answer 1

# coding=utf-8
# 如果你的意思是转化为array数组的话:
import MySQLdb as mdb
import numpy as np
import sys

try:
    ip = '数据库ip'
    user = '数据库用户名'
    password = '密码'
    dbname = '你想查询的数据库名'
    # 新建一个连接到数据库
    con = mdb.connect(ip, user, password, dbname)
    with con:
        cur = con.cursor(mdb.cursors.DictCursor)
        # 执行查询
        cur.execute("select * from table limit 100;")
        # 将查询结果存入变量rows
        rows = cur.fetchall()
        # 将rows转化为数组
        rows = np.array(rows)

finally:
    if con:
        con.close()  # 无论如何, 记得关闭连接