scipy ndimage.rotate不能使用np.nan值

Question

我有一個2維numpy數組，numpy數組中的某處有np.nan值。 現在我試圖旋轉90度的陣列。但是，scipy.ndimage.rotate不起作用。

下面是一個簡單的測試：

import numpy as np 
from scipy import ndimage 

a = np.array([[1, 2,], [3, 4]]) 
b = np.array([[np.nan, 2], [3, 4]]) 

ndimage.rotate(a, 90) 
# result: 
# array([[2, 4], 
#  [1, 3]]) 

ndimage.rotate(b, 90) 
# result: 
# array([[ nan, nan], 
#  [ nan, nan]]) 

有沒有辦法解決ndimage旋轉的問題呢？

Answer 1

你可以使用numpy.rot90：

np.rot90(b) 

[[ 2. 4.] 
[ nan 3.]] 

對於爲什麼scipy.ndimage.rotate不工作的說明，請參見Evert's answer。

Answer 2

scipy.ndimage.rotate實際上是scipy.ndimage.interpolation.rotate，它表示問題：數組中的值被插值，並且用NaN插值任何東西都會導致NaN（同上無窮大）。

顯然，加布裏埃爾建議的rot90只是切換元素，而不需要插值。通用rotate需要插值，因爲例如通過旋轉。 42度不是簡單的元素變化。

總而言之，我會說rotate確實工作與NaN。只是因爲上述原因，不符合你的預期。