MySQL Python 的 PHP mysql_fetch_array 等效方法

在本文中，我们将介绍如何在 Python 中实现 PHP 中的 mysql_fetch_array 方法的功能。

阅读更多：MySQL 教程

什么是 mysql_fetch_array

mysql_fetch_array 是 PHP 中的一个数据库函数，用于从 MySQL 查询语句中获取行数据。该函数可以返回一个关联数组、一个数字索引数组、或者两者兼有的数组。比如下面的 PHP 语句，将返回一个关联数组和一个数字索引数组：

$result = mysql_query("SELECT * FROM my_table");
$row_assoc = mysql_fetch_array($result, MYSQL_ASSOC);
$row_num = mysql_fetch_array($result, MYSQL_NUM);

MySQL Python 中的等效实现

在 Python 中，我们可以使用 MySQLdb 或 PyMySQL 模块来连接 MySQL 数据库。同样，我们可以使用 cursor 对象的 fetchone() 或 fetchall() 方法来获取查询结果，然后根据需要将其转化为字典或元组。

fetchone() 方法

fetchone() 方法将返回结果集中的下一行，即一个包含行数据的元组。每一次调用 fetchone() 方法，都会获取结果集中的下一行，直到获取完所有行。如果没有更多的行，则返回 None。具体实现如下：

import MySQLdb

conn = MySQLdb.connect(host="localhost", user="root", passwd="", db="mydatabase")
cursor = conn.cursor()

cursor.execute("SELECT * FROM my_table")
row = cursor.fetchone()
while row is not None:
    # do something with row
    row = cursor.fetchone()

对于上述代码，我们可以封装成一个函数，使其能够像 mysql_fetch_array 一样，返回关联数组或数字索引数组：

def mysql_fetch_array(cursor, result_type):
    cursor.execute("SELECT * FROM my_table")
    row = cursor.fetchone()
    res = []
    while row is not None:
        if result_type == MYSQL_ASSOC:
            res.append(dict(zip(cursor.column_names, row)))
        elif result_type == MYSQL_NUM:
            res.append(row)
        row = cursor.fetchone()
    return res

fetchall() 方法

fetchall() 方法将返回结果集中所有的行，即一个包含所有行数据的元组列表。本质上，它相当于多次调用 fetchone() 方法，直到获取完所有行。具体实现如下：

import MySQLdb

conn = MySQLdb.connect(host="localhost", user="root", passwd="", db="mydatabase")
cursor = conn.cursor()

cursor.execute("SELECT * FROM my_table")
rows = cursor.fetchall()
for row in rows:
    # do something with row

同样，我们也可以封装成一个函数，来实现 mysql_fetch_array 的功能：

def mysql_fetch_array(cursor, result_type):
    cursor.execute("SELECT * FROM my_table")
    rows = cursor.fetchall()
    res = []
    for row in rows:
        if result_type == MYSQL_ASSOC:
            res.append(dict(zip(cursor.column_names, row)))
        elif result_type == MYSQL_NUM:
            res.append(row)
    return res

总结

本文介绍了在 MySQL Python 中实现 PHP 中 mysql_fetch_array 的等效方法。你可以使用 fetchone() 方法获取一行数据，或使用 fetchall() 方法获取所有数据。如果你需要返回关联数组或数字索引数组，你可以根据需要封装一下代码，将结果转化为字典或元组。在 MySQL Python 中，你可以使用 MySQLdb 或 PyMySQL 模块来连接 MySQL 数据库。