python矩阵输出对齐_Python：对齐NumPy数组

小编典典

这里有一个量化的方法，通过启发this other post和推广到覆盖non-zeros所有四个方向-

def justify(a, invalid_val=0, axis=1, side='left'):

"""

Justifies a 2D array

Parameters

----------

A : ndarray

Input array to be justified

axis : int

Axis along which justification is to be made

side : str

Direction of justification. It could be 'left', 'right', 'up', 'down'

It should be 'left' or 'right' for axis=1 and 'up' or 'down' for axis=0.

"""

if invalid_val is np.nan:

mask = ~np.isnan(a)

else:

mask = a!=invalid_val

justified_mask = np.sort(mask,axis=axis)

if (side=='up') | (side=='left'):

justified_mask = np.flip(justified_ma