我曾经是MATLAB的用户,现在正在转向Python。
我试图以简洁的方式重写以下MATLAB代码,将其转换为Python:
s = sum(Mtx);
newMtx = Mtx(:, s>0);
其中,Mtx是一个二维稀疏矩阵。
我的Python解决方案如下:
s = Mtx.sum(0)
newMtx = Mtx[:, np.where((s>0).flat)[0]] # taking the columns with nonzero indices
Mtx是一个二维的CSC稀疏矩阵。
这段Python代码不如Matlab那么易读/优雅。有什么好的想法可以更加优雅地编写它吗?
谢谢!