Question

How do I trim extra zeros from a 2D array in python?

Answer 1

Hey @Laksha, you can try something like this:

>>> np.array([sub[~(sub == 0)].tolist() for sub in a if sub[sub != 0].tolist()])
array([[4, 1, 2], [3, 6]], dtype=object)
>>>