如何动态构建Python列表推导式以适配多API数据提取需求？

针对你在AWS Lambda中需要根据不同API结构动态提取数据字段的需求，这里提供两种实用方案，避开动态生成列表推导式的复杂操作，同时兼顾安全性和灵活性：

方案一：字段路径解析（推荐，安全可控） #

通过定义字符串格式的字段路径来描述嵌套字段的位置，再用通用函数解析路径提取值，无需修改列表推导式的结构。

实现步骤： #

1. 编写嵌套字段提取函数

def get_nested_value(obj, path):
    """根据字符串路径提取嵌套对象的值，支持数组索引（如'orders.0.status'）"""
    keys = path.split('.')
    for key in keys:
        # 处理数组索引，将字符串数字转为整数
        if key.isdigit():
            key = int(key)
        obj = obj[key]
    return obj

2. 动态传入字段路径并提取数据
   在Lambda handler中，可从事件参数或环境变量获取字段路径列表，直接用于数据提取：

import requests
import csv
from io import StringIO
import boto3

def get_nested_value(obj, path):
    keys = path.split('.')
    for key in keys:
        if key.isdigit():
            key = int(key)
        obj = obj[key]
    return obj

def lambda_handler(event, context):
    # 从事件参数获取输入：URL、请求头、参数、字段路径
    url = event['url']
    headers = event.get('headers', {})
    params = event.get('params', {})
    field_paths = event['field_paths']  # 示例：['id','name','address'] 或 ['orders.0.status','assignee.id']
    
    # 调用API获取数据
    response = requests.get(url, headers=headers, params=params)
    response.raise_for_status()
    records = response.json()
    
    # 动态提取目标字段
    values = [tuple(get_nested_value(x, path) for path in field_paths) for x in records]
    
    # 写入CSV并上传到S3（Lambda常见场景）
    output = StringIO()
    writer = csv.writer(output)
    writer.writerow(field_paths)  # 用路径作为表头
    writer.writerows(values)
    
    s3 = boto3.client('s3')
    s3.put_object(
        Bucket='your-bucket-name',
        Key=event.get('csv_filename', 'output.csv'),
        Body=output.getvalue(),
        ContentType='text/csv'
    )
    
    return {'statusCode': 200, 'body': 'CSV生成并上传成功'}

这种方式的优势：无需动态生成代码，避免注入风险，字段路径易配置（可存于配置文件或通过事件参数传递）。

方案二：动态Lambda表达式（仅可信输入场景使用） #

如果必须使用类似列表推导式的表达式语法，可通过eval动态生成提取函数，但仅适用于输入完全可信的场景（否则存在代码注入风险）。

示例代码：

def lambda_handler(event, context):
    # 从事件中获取提取表达式，示例："(x['id'], x['name'], x['address'])"
    extract_expr = event['extract_expr']
    # 动态生成提取函数
    extract_func = eval(f"lambda x: {extract_expr}")
    
    # 获取API数据
    response = requests.get(event['url'], headers=event.get('headers', {}), params=event.get('params', {}))
    records = response.json()
    
    # 提取数据
    values = [extract_func(x) for x in records]
    
    # 后续CSV写入逻辑同方案一...

警告：若extract_expr来自外部不可信输入，攻击者可注入恶意代码执行，严重威胁Lambda环境安全，因此不推荐在生产环境使用。

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