2d dizisinde Bazı Nonzero Elemanı Olan Sütunların Endekslerini Bul

Question

Loşlu bir numpy dizim var (157,1944).

Herhangi bir sırada Nonzero öğesi olan sütunların indislerini almak istiyorum.

örnek: [[0,0,3,4], [0,0,1,1]] ---->[2,3]

her satırı bakarsanız, sütunlarda Olmayan Sıfır bileşen olduğuna [2, 3] ben var ise

[[0,1,3,4], [0,0,1,1]]

[1,2,3] numaralı telefonu almalıyım çünkü sütun dizini 0 herhangi bir sırada Nonzero öğesi içermiyor.

Answer 1

Sorunuzun tamamen tanımlanmış olduğundan emin değilsiniz. Ancak, satır hiçbiri sıfırdan farklı olduğu için sütunların endeksleri olan

import numpy as np 
a = np.array([[0,0,3,4], [0,0,1,1]]) 

sonra

>>> np.nonzero(np.all(a != 0, axis=0))[0] 
array([2, 3]) 

başladığınızı düşünelim ve

>>> np.nonzero(np.any(a != 0, axis=0))[0] 
array([2, 3]) 

endeksleri olan Satırların numaralı sütununun sıfır olmadığı sütunlar (verdiğiniz örnekle aynı olur).