如何在Python中自动将for循环转换为while循环？

基于AST的工具与自定义实现 #

既然你已经尝试过Python原生AST库，以下几种方案可以满足你的需求：

1. 基于lib2to3自定义转换规则 #

lib2to3是Python官方提供的代码转换工具，原本用于版本迁移，但可以自定义fixer实现for到while的转换。你可以编写一个fixer类，匹配for var in range(n)这种模式的AST节点，然后将其替换为对应的变量初始化、while循环以及自增语句的组合。

2. 用rope做代码重构 #

rope是专注于Python代码重构的库，它提供了便捷的API来修改代码结构。你可以通过它定位目标for循环节点，然后直接替换为等价的while循环结构，它会自动处理变量作用域等细节，无需手动遍历AST。

3. 原生AST自定义脚本（无需第三方库） #

如果想自己实现，基于原生AST完全可以做到，以下是一个极简示例：

import ast
import astor  # Python3.9+也可以用ast.unparse

def transform_for_to_while(node):
    # 处理for循环节点
    if isinstance(node, ast.For):
        # 仅匹配for var in range(n)的简单场景
        if (isinstance(node.iter, ast.Call) and 
            isinstance(node.iter.func, ast.Name) and 
            node.iter.func.id == 'range' and 
            len(node.iter.args) == 1):
            # 构造变量初始化语句：var = 0
            assign_stmt = ast.Assign(
                targets=[node.target],
                value=ast.Constant(value=0)
            )
            # 构造while条件：var < n
            while_test = ast.Compare(
                left=node.target,
                ops=[ast.Lt()],
                comparators=node.iter.args
            )
            # 构造自增语句：var +=1
            increment_stmt = ast.AugAssign(
                target=node.target,
                op=ast.Add(),
                value=ast.Constant(value=1)
            )
            # 将自增加入循环体末尾
            node.body.append(increment_stmt)
            # 返回初始化语句和while循环
            return [assign_stmt, ast.While(test=while_test, body=node.body, orelse=node.orelse)]
    
    # 递归处理子节点
    if hasattr(node, 'body'):
        updated_body = []
        for child in node.body:
            transformed = transform_for_to_while(child)
            if isinstance(transformed, list):
                updated_body.extend(transformed)
            else:
                updated_body.append(transformed)
        node.body = updated_body
    return node

# 测试转换
source_code = """
for i in range(10):
    print(i)
"""
ast_tree = ast.parse(source_code)
modified_tree = transform_for_to_while(ast_tree)
print(astor.to_source(modified_tree))

这个脚本可以将你示例中的for循环转换为目标while结构，你还可以扩展逻辑来支持range(start, stop, step)这类更复杂的场景。

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