获取numpy矩阵的前n个非零元素答案

【问题标题】：Getting first n non-zero elements of a numpy matrix获取numpy矩阵的前n个非零元素
【发布时间】：2017-02-16 17:11:48
【问题描述】：

我有一个这样的矩阵 M：

>>> M
array([[1, 0, 3, 4],
       [0, 3, 4, 5],
       [5, 4, 0, 7]])

我要做的是提取M中每一行的前N个（假设N = 2）非零元素并将它们放入相同形状的新矩阵M2中，同时将匹配值设置为米为零。所以输出应该是：

>>> M
array([[0, 0, 0, 4],
       [0, 0, 0, 5],
       [0, 0, 0, 7]])

>>> M2
array([[1, 0, 3, 0],
       [0, 3, 4, 0],
       [5, 4, 0, 0]])

【问题讨论】：

删除您的recent question 的做法非常糟糕，特别是考虑到有人花时间清理问题中的不足之处，然后实际解决问题。考虑到删除问题的行为，人们会发现回答您的问题的动力较小。

标签： python numpy

【解决方案1】：

cumsum 的一种方法-

M2 = M*((M!=0).cumsum(1)<=2)
M_new = M - M2

示例运行 -

In [42]: M
Out[42]: 
array([[1, 0, 3, 4],
       [0, 3, 4, 5],
       [5, 4, 0, 7]])

In [43]: M2 = M*((M!=0).cumsum(1)<=2)
    ...: M_new = M - M2
    ...: 

In [44]: M_new
Out[44]: 
array([[0, 0, 0, 4],
       [0, 0, 0, 5],
       [0, 0, 0, 7]])

In [45]: M2
Out[45]: 
array([[1, 0, 3, 0],
       [0, 3, 4, 0],
       [5, 4, 0, 0]])

【讨论】：

我真的很喜欢你的推理！干得好！