如何在python源代码中找到给定变量的导入模块依赖项

import ast

def find_import_dependencies(source_code, variable_name):
    dependencies = []

    # 解析源代码为抽象语法树
    tree = ast.parse(source_code)

    # 遍历抽象语法树
    for node in ast.walk(tree):
        if isinstance(node, ast.Import):
            # 处理import语句
            for alias in node.names:
                if variable_name in alias.asname or variable_name in alias.name:
                    dependencies.append(alias.name)
        elif isinstance(node, ast.ImportFrom):
            # 处理from ... import ...语句
            if node.module:
                module_name = node.module
                for alias in node.names:
                    if variable_name in alias.asname or variable_name in alias.name:
                        dependencies.append(module_name + '.' + alias.name)

    return dependencies

# 示例源代码
source_code = '''
import math
from datetime import datetime
import numpy as np
from sklearn.linear_model import LinearRegression as LR
'''

# 给定变量名
variable_name = 'LR'

# 查找导入模块依赖项
dependencies = find_import_dependencies(source_code, variable_name)

# 输出结果
print(dependencies)

['sklearn.linear_model.LinearRegression']