更改列中第一個非零元素上方的數組值

Question

我正在尋找矢量化方法來更改列中第一個非零元素上方的數組值。

for x in range(array.shape[1]):
    for y in range(array.shape[0]):
        if array[y,x]>0:
            break
        else:
            array[y,x]=255

在

出去

Answer 1

正如您寫的關於數組（不是DataFrame ），我假設您有一個Numpy數組並且想要使用Numpy方法。

要完成您的任務，請運行以下代碼：

np.where(np.cumsum(np.not_equal(array, 0), axis=0), array, 255)

步驟示例及說明：

1. 源數組：

    array([[0, 1, 0], [0, 0, 1], [1, 1, 0], [1, 0, 0]])

2. np.not_equal(array, 0)計算一個 boolean 數組，元素為True ：= 0：

    array([[False, True, False], [False, False, True], [ True, True, False], [ True, False, False]])

3. np.cumsum(..., axis=0) ) 沿軸0 （列）計算累積和（ True計為1 ）：

    array([[0, 1, 0], [0, 1, 1], [1, 2, 1], [2, 2, 1]], dtype=int32)

4. 上面的數組是where 中使用的掩碼。 對於掩碼值（其中掩碼的對應元素為True （實際上， != 0 ）），從數組的對應元素中取值，否則取255 ：

    np.where(..., array, 255)

結果（對於我的數組）是：

array([[255,   1, 255],
       [255,   0,   1],
       [  1,   1,   0],
       [  1,   0,   0]])

Answer 2

使用遮罩：

array[array == 0] = 255