取出mask位置所在的所有元素

最新推荐文章于 2024-02-20 18:48:48 发布

<paradox.h>

最新推荐文章于 2024-02-20 18:48:48 发布

阅读量304

点赞数 1

文章标签： numpy python

本文链接：https://blog.csdn.net/xcxcxcxc12321/article/details/127598617

版权

import numpy as np
a = np.array([[22, 31, 25, 65],
              [24, 53, 64, 53],
              [22, 87, 88, 85],
              [52, 34, 23, 76]])
mask = np.array([[1, 1, 1, 0],
                 [0, 1, 1, 0],
                 [0, 0, 0, 0],
                 [0, 0, 1, 0]])
print(np.where(mask == 1))
# output: (array([0, 0, 0, 1, 1, 3], dtype=int64), array([0, 1, 2, 1, 2, 2], dtype=int64))

print(a[np.where(mask == 1)])
# output: [22 31 25 53 64 23]