如何在Python中为列表每个元素应用含双变量的自定义函数？

嘿，作为Python新手，能尝试自己写代码已经很棒啦！咱们先看看你代码里的问题，再一步步修正实现你想要的功能～

原代码的问题分析 #

你的思路是对的，但有几个关键的小问题导致代码无法工作：

• weight函数里初始化了空的return_list，然后循环这个空列表，代码根本不会执行循环体，最后返回的还是空列表；
• 没有把database里的元素和对应的speed、time参数关联起来，也没把database传递给处理函数；
• run函数只定义了database，没有完成后续的计算逻辑，也没调用处理函数。

修正后的实现方案 #

首先，我们需要给每个元素（A、B、C）绑定对应的speed和time参数，然后遍历database，对每个元素调用weighting函数计算结果，最后收集成新数组。

完整代码如下：

# 定义你的自定义计算函数
def weighting(speed, time):
    return speed * time

def run():
    database = ['A', 'B', 'A', 'B', 'C']
    # 建立元素与对应参数的映射，你可以根据需求修改这里的数值
    element_params = {
        'A': (5, 2),  # A对应的speed=5，time=2
        'B': (3, 4),  # B对应的speed=3，time=4
        'C': (6, 1)   # C对应的speed=6，time=1
    }
    
    # 生成计算后的新数组
    result = []
    for element in database:
        # 取出当前元素对应的speed和time
        speed, time = element_params[element]
        # 调用计算函数并把结果加入列表
        result.append(weighting(speed, time))
    
    print("计算后的新数组：", result)

# 执行run函数
run()

更简洁的写法（列表推导式） #

如果你想让代码更Pythonic，可以用列表推导式简化循环部分：

def run():
    database = ['A', 'B', 'A', 'B', 'C']
    element_params = {'A': (5,2), 'B':(3,4), 'C':(6,1)}
    # 一行代码生成结果
    result = [weighting(*element_params[elem]) for elem in database]
    print("计算后的新数组：", result)

解释一下关键部分 #

• element_params是一个字典，用来存储每个元素对应的(speed, time)参数对，这样我们能快速找到每个元素需要的参数；
• 循环遍历database时，每次取出元素对应的参数，调用weighting函数计算，把结果添加到result列表中；
• 列表推导式是Python里很实用的语法，能快速生成列表，让代码更简洁。

如果需要调整参数或者有其他疑问，随时问哦～

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