将输出函数附加到多个列表

Question

我想执行一个具有不同参数值的函数。我有以下运行良好的代码 sn-p：

tau = np.arange(2,4.01,0.1)
R = []
P = []
T = []
L = []
D = []
E = []
Obj = []
for i, tenum in enumerate(tau):
    [r, p, t, l, d, e, obj] = (foo.cvxEDA(edaN, 1./fs, tenum, 0.7, 10.0, 0.0008, 0.01))
    R.append(r)
    P.append(p)
    T.append(t)
    L.append(l)
    D.append(d)
    E.append(e)
    Obj.append(obj)

然而，我想知道：有没有更简单的方法来完成这个？

---

我尝试过使用 res.append(foo.cvxEDA(edaN, 1./fs, tenum, 0.7, 10.0, 0.0008, 0.01) 但res[1] 返回<generator object <genexpr> at 0x046E7698>。

Answer 1

tau = np.arange(2,4.01,0.1)
results = [[] for _ in range(7)]
for i, tenum in enumerate(tau):
    data = foo.cvxEDA(edaN, 1./fs, tenum, 0.7, 10.0, 0.0008, 0.01)
    for r,d in zip(results, data):
        r.append(d)
r, p, t, l, d, e, _obj = results

Answer 2

您可以将生成器对象转换为列表对象，只需将其传递给list() 函数，这样也许可以满足您的需求：

res = []
for i, tenum in enumerate(tau):
   res.append(list(foo.cvxEDA(edaN, 1./fs, tenum, 0.7, 10.0, 0.0008, 0.01)))

列表理解更短：

res = [list(foo.cvxEDA(edaN, 1./fs, tenum, 0.7, 10.0, 0.0008, 0.01)) for i, tenum in enumerate(tau)]

无论哪种方式，与您想要的相比，这都会使 res 转置（将其视为矩阵）。你可以通过拨打zip来解决这个问题：

res_tr = zip(*res)
R, P, T, L, D, E, Obj = res_tr

编辑：最简单的是，您可以避免使用直接传递给zip() 的生成器表达式来构建中间列表：

R, P, T, L, D, E, Obj = zip(*(list(foo.cvxEDA(edaN, 1./fs, tenum, 0.7, 10.0, 0.0008, 0.01)) for tenum in tau))

最后一点：在所有这些中，您可以将“for i, tenum in enumerate(tau)”替换为“for tenum in tau”，因为您似乎没有使用i。