Numpy裁剪2D陣列到非NaN值

Question

我有一個2D Numpy數組，其中包含一個被NaN海洋包圍的數據島。

我想裁剪數組，使其只包含邊界框和非NaN區域的內容。

我還想知道這個盒子的索引和大小與原始數組有關。

這可能嗎？

Answer 1

import numpy as np
a = np.empty((15,10))
a.fill(np.nan)
a[7,6] = 76
a[8,5] = 85
a[9,5] = 95
a[9,7] = 97

現在我們有了一個由15行和10列組成的數組，其中包含NAN海洋中的一個數據島：

nan,  76.,  nan,
85.,  nan,  nan,
95.,  nan,  97.,

下一個：

nans = np.isnan(a)
nancols = np.all(nans, axis=0) # 10 booleans, True where col is all NAN
nanrows = np.all(nans, axis=1) # 15 booleans

firstcol = nancols.argmin() # 5, the first index where not NAN
firstrow = nanrows.argmin() # 7

lastcol = len(nancols) - nancols[::-1].argmin() # 8, last index where not NAN
lastrow = len(nanrows) - nanrows[::-1].argmin() # 10

最后：

a[firstrow:lastrow,firstcol:lastcol]

向我們展示了這個島嶼：

array([[ nan,  76.,  nan],
       [ 85.,  nan,  nan],
       [ 95.,  nan,  97.]])