python np.arange 步长0.1的问题需要注意

两个小测试：

1）count输出什么？

import numpy as np
    y_min= 20
    y_max= 21
    step = 0.1
    count = 0

    for i in np.arange(y_min, y_max, step):
        count += 1
        print('i:', count, i)

答案是：count = 10

如图

 

2）count输出是什么？

import numpy as np
    y_min= 20
    y_max= 21.1
    step = 0.1
    count = 0

    for i in np.arange(y_min, y_max, step):
        count += 1
        print('i:', count, i)

答案是：count = 12

如图

结论：python 的np.arange的步长默认是整数，对应范围是前闭后开[start, end)，但是，步长是小数时，就不适用了，很容易出错，需要注意。