Python 中的 MATLAB arrayfun 替代方案

Question

在 MATLAB 中，我执行如下操作：

z = diag(c)*D*x;
idxN = z<0; idxP = z>=0; % logical indexing
y1 = sum(-z(idxN))+sum(arrayfun(@(x) log(exp(x)+1),z(idxN))); 

我正在尝试编写一个 Python：

z = np.diagflat(c).dot(D).dot(x)
idxN = z<0, idxP = z>=0
y1 = np.sum(-z[idxN])+np.sum(math.log(np.exp(x)+1),z[idxN]); 

上面的 sn-p 不起作用，因为我猜我必须在某个地方矢量化？

感谢您的意见。

Answer 1

arrayfun 是 MATLAB 中循环的单行包装器。在 Python 中，你可以这样做：

y1 = np.sum(-z[idxN]) + np.sum([math.log(np.exp(x)+1) for x in z[idxN]])