numpy布尔数组索引

    xiaoxiao2022-07-07  205

    1.先创建一个数组如下

    h=np.arange(6).reshape(3,2)

    2.选择出大于等于3的数据,会得到一个布尔类型的数组,如果大于等于3 ,在bool_idx数组中就会展示True,否则为Flase。如图:

    bool_idx=(h>=3)

    3.打印出h数组中大于等于3的数据

    print(h[bool_idx])

    4.也可以直接把条件和数组结合:

    print(h(h>=3))

    也会得到上面的结果。

    最新回复(0)