Erste erste n Nicht-Null-Elemente eines numpy Matrix

Question

ich eine Matrix M wie dieses:

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

Was ich tun möchte, ist die erste N zu extrahieren ist (sagen wir N = 2) nicht Null Elemente jeder Reihe in M ​​und setze sie in eine neue Matrix M2 der gleichen Form, während die übereinstimmenden Werte in M ​​auf Null gesetzt werden. So soll die Ausgabe sein:

>>> 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]]) 

Answer 1

Einen Ansatz mit cumsum -

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

Probelauf -

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]])