https://numpy.org/doc/stable/reference/generated/numpy.where.html
python（九）：np.where
原文：https://www.cnblogs.com/zhangxianrong/p/14930517.html
