如何在 True 的同时为每层 3D 添加 2D

Question

我有 3 个数组。

a = [[0 1] [2 3]]
b = [[True False] [True False]]
c = [[[1 1] [1 1] [1 1] [1 1]][[1 1] [1 1] [1 1] [1 1]]]

a是包含数字的二维数组。b是包含真或假语句的二维数组。c是包含数字的 3D 数组。

假设c有 2 层 2D 数组，我想在每一层c中添加a，只要b为 True。有没有办法在不对 python 中的每个单元格使用循环的情况下做到这一点？

预期成果

k = [[2 1] [4 3]]

Answer 1

谢谢你的回答。我想我找到了我想要的答案。

k = np.where(b,a+c.sum(axis = 1),a+0).