我想使用列表作为参数，从函数的结果生成一个列表

Question

我试图通过使用其他列表作为函数的参数来创建一个列表。但是，我似乎无法获得正确的语法。
这是我的代码：

f1 = theta0 + theta1*(X1_train) + theta2*(X2_train)+theta3*(X3_train)

预期结果将是一份长度相同的X1_序列、X2_序列和X3_序列的列表（这三个序列的长度相同）

我希望得到列表X1\u列、X2\u列和X3\u列上每个元素的结果列表，作为函数的参数。例如，如果我的列表是

X1_train = [0, 1]
X2_train = [1, 2]
X3_train = [0, 2]

我希望有一个数字列表，比如

f1 = [theta0 + theta2, theta0 + theta1 + theta2 + 2*theta3]

θ是随机数

这个列表是一个数据帧的列，我将其转换为列表，这样我就可以完成这个功能

Answer 1

我建议您尝试以下简单代码：

def f(X: tuple, theta: tuple):
   if not isinstance(X, tuple):
      raise TypeError('X must be a tuple')
   if not isinstance(theta, tuple):
      raise TypeError('theta must be a tuple')
   if not X.__len__() == theta.__len__():
      raise ValueError('length of tuples is not equal')
   return sum([np.array(x_)*t_ for x_, t_ in zip(X, theta)])

注意，如果X或θ不是相同长度的元组，则会抛出错误

示例：（应该按原样工作

import numpy as np
X1_train = [0, 1]
X2_train = [1, 2]
X3_train = [0, 2]

theta_1 = 1
theta_2 = 1
theta_3 = 3

print(f(
    (X1_train, X2_train, X3_train),
    (theta_1, theta_2, theta_3)
    ))
>>> [1 9]