Numpy中bool式条件索引【Python琐碎知识点】

最新推荐文章于 2024-05-22 01:13:49 发布

MarToony|名角

最新推荐文章于 2024-05-22 01:13:49 发布

阅读量962

点赞数

分类专栏： Python琐碎知识点文章标签： python numpy 索引

本文链接：https://blog.csdn.net/m0_38052500/article/details/106858787

版权

Python琐碎知识点专栏收录该内容

6 篇文章 1 订阅

订阅专栏

a  = np.arange(24).reshape(2,3,4)
a
"""
array([[[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]],

       [[12, 13, 14, 15],
        [16, 17, 18, 19],
        [20, 21, 22, 23]]])
"""

[a>13]
"""
[array([[[False, False, False, False],
         [False, False, False, False],
         [False, False, False, False]],
 
        [[False, False,  True,  True],
         [ True,  True,  True,  True],
         [ True,  True,  True,  True]]])]
"""
# 单个条件时，返回一维数组，而非原来的shape
a[a>13]
# 两个条件时
a[(a>15)&(a<20)]

############ 输出内容 ########### 
array([14, 15, 16, 17, 18, 19, 20, 21, 22, 23])
array([16, 17, 18, 19])

需要提醒的是，如果仅限于条件判断，会返回原来相同shape的矩阵；而如果条件判断的shape矩阵作为索引的方式，则仅会返回一个一维数组。