天天看點

NumPy數組切片的複制問題

在使用NumPy數組時,有一個要注意的地方:在取數組的切片時,取出來的切片(Slices)僅僅是原始數組的視圖(Views),而非它的複制!這與Python的built-in的list不同。

arr = np.arange(10)
arr      

輸出:array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

接着,可以用廣播的方式給其中的切片指派:

arr[5:8] = 12
arr      

輸出:array([ 0,  1,  2,  3,  4, 12, 12, 12,  8,  9])

如果我們用切片的方式,将其切片取出後再指派:

arr_slice = arr[5:8]
arr_slice[1] = 12345
arr      

輸出:array([    0,     1,     2,     3,     4,    12, 12345,    12,     8,   9])

可以看到,原始數組的值也發生了改變。

如果想得到ndarray切片的副本,而非視圖,那就要顯式地複制,如下:

arr_slice_copy = arr[5:8].copy()
arr_slice_copy[:] = 0
arr