如何用Python从SQL每行取值传入函数后更新对应行数据

方案一：纯SQL批量更新（推荐，效率更高） #

直接用数据库内置函数计算年龄并批量更新，无需Python循环，适合数据量大的场景。

不同数据库的SQL语句： #

• MySQL/MariaDB：

UPDATE employees 
SET age = TIMESTAMPDIFF(YEAR, birthdate, CURDATE())
WHERE (age IS NULL OR age = '') AND birthdate != '';

• PostgreSQL：

UPDATE employees 
SET age = EXTRACT(YEAR FROM AGE(CURRENT_DATE, birthdate))::INT
WHERE (age IS NULL OR age = '') AND birthdate IS NOT NULL;

• SQL Server：

UPDATE employees 
SET age = DATEDIFF(YEAR, birthdate, GETDATE())
WHERE (age IS NULL OR age = '') AND birthdate IS NOT NULL;

说明：

• 数据库自带的日期函数计算年龄，比Python循环效率高很多
• WHERE条件精准过滤：未填写年龄（空值或空字符串）且有有效出生日期的行
• 执行后记得提交事务（部分数据库需手动执行）

方案二：Python循环处理（适合复杂自定义逻辑） #

如果需要自定义年龄计算规则（比如特殊生日判断），可以用Python逐行处理，但要修正你原代码的致命错误：必须用主键定位单行，否则会把所有未填年龄的行改成同一个值。

修正后的代码：

import datetime
# 替换成你的数据库连接库，比如pymysql/psycopg2等
import pymysql

# 建立数据库连接
connection = pymysql.connect(
    host='你的数据库地址',
    user='用户名',
    password='密码',
    database='数据库名'
)

cursor = connection.cursor()

# 只查询需要更新的行：获取主键和出生日期，避免SELECT *浪费资源
cursor.execute("""
    SELECT employee_id, birthdate 
    FROM employees 
    WHERE (age IS NULL OR age = '') AND birthdate != ''
""")
rows = cursor.fetchall()

today = datetime.datetime.now()
current_year = today.year

for employee_id, birthdate in rows:
    # 计算年龄：处理生日未过的情况，年龄减1
    birth_year = birthdate.year
    birth_month = birthdate.month
    birth_day = birthdate.day
    
    age = current_year - birth_year
    if (today.month, today.day) < (birth_month, birth_day):
        age -= 1
    
    # 用主键employee_id定位单行更新，避免批量错误
    update_query = "UPDATE employees SET age = %s WHERE employee_id = %s"
    cursor.execute(update_query, (age, employee_id))

# 必须提交事务，否则更新不会生效
connection.commit()

# 关闭连接
cursor.close()
connection.close()

关键修正点：

1. 只查询必要字段（主键+出生日期），提升查询效率
2. 更新时用主键employee_id定位单行，解决原代码中WHERE age = ''导致的批量错误更新问题
3. 增加生日未过的年龄修正逻辑，计算结果更准确
4. 最后必须执行connection.commit()提交事务

内容的提问来源于stack exchange，提问作者Sean