如何在Scipy中將圖像加載爲具有閾值的二值圖像？

Question

我有一個灰度圖像 - MNIST數據集，實際上 - 我需要將其轉換爲閾值爲240的二值圖像，以便所有低於240的值都是1，並且上面的所有值都是零。

這是matlab中的一個函數，所以我確定在scipy中有一些相應的函數......但它避開了我的搜索。

---

等效地，如果我有一個（60000,28,28）形ndarray，如何我有條件地檢查所有的值和設置值高於240到零，其餘的爲1的？

在僞代碼numpy的，

image_array = big_array_of_28x28_images 

bw_image_array = image_array[image_array > 240 yield 0, else yield 1] 

Answer 1

如果A是您的矩陣，二進制矩陣B爲：

B = np.where（A < = 240，1，0）