如何将SQL语句的结果保存到变量中？

import mysql.connector

# 连接到MySQL数据库
cnx = mysql.connector.connect(user='username', password='password',
                              host='localhost', database='database_name')

# 创建游标对象
cursor = cnx.cursor()

# 执行SQL查询语句
query = "SELECT column_name FROM table_name WHERE condition"
cursor.execute(query)

# 获取结果集中的数据
result = cursor.fetchone()  # 获取一行数据

# 将数据保存到变量中
variable = result[0]  # 假设结果集只有一列

# 关闭游标和数据库连接
cursor.close()
cnx.close()