Python根据列表的内容评估函数

Question

我继承了一个有效的代码列表:

A = [ 's1', 's2', 's3', 's1' ] #the actual size is about 200
B = [ 1, 2, 13, 4 ] #same size as A

s1、s2、s3 是动态定义的变量:

s1 = 5
s2 = 3
s3 = 13

我有一个函数定义为:

def fun1( s, arg2 ):
 return s * numpy.random.normal( 0, 1, ( 200, arg2 ) )

上面函数中的arg2来自选中的s1对应的数组B

我想生成这样的 C:

C = [ fun1(s1,1), fun1(s2,2), fun1(s3,13), fun1(s1,4) ] #C can be list or collection of arrays, not sure about the best data structure for C

Answer 1

在使用列表理解之前，使用内置函数 zip() 将 A 和 B 组合成一个元组列表:

import numpy as np

s1 = 5
s2 = 3
s3 = 13

A = ['s1', 's2', 's3', 's1']
B = [1, 2, 13, 4]

def fun1(s, arg2):
 return s * np.random.normal(0, 1, (200, arg2))

C = [fun1(locals()[A_item], B_item) for A_item, B_item in zip(A, B)]