如何在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）