用于列表理解的嵌套循环

如何解决用于列表理解的嵌套循环

我必须将嵌套循环转换为列表理解：

H = cvxopt.matrix(0.0,(rows,rows),tc='d')

for i in range(rows):
    for j in range(rows):
        H[i,j] = t[i] * t[j] * kf(X[i],X[j],param)

我已经试过了：

H = [t[i] * t[j] * kf(X[i],param) for H[i,j] in H]

但这给了我一个错误 i 未定义。

解决方法

您不能完全从列表推导式中构建矩阵，只能使用 lists。那只是因为 Python 没有内置矩阵类型。

您可以构建下一个最好的东西，list 的 list。例如

>>> [[i + 3 * j for i in range(N)] for j in range(N)]
[[0,1,2],[3,4,5],[6,7,8]]

然后您可以使用 cvxopt.matrix 构造函数转换为 matrix。

H = cvxopt.matrix([[i + 3 * j for i in range(N)] for j in range(N)])

在您的示例中，t 被定义在某处（问题中未包含），您可以这样做：

H = cvxopt.matrix([[t[i] * t[j] * kf(X[i],X[j],param) for i in range(rows)] for j in range(rows)])