Python 如何用另一个numpy数组屏蔽numpy数组？

如何使用numpy内置（no

for

loop）从

x

和

z

生成输出？

x*z[…，None]

？@Divakar，谢谢。如果有，请提供此类操作的文档，这可能是一个良好的开端。另外，还可以查找None/np.newaxis。

x*np.dstack（（z，z，z））

也可以，但是@Divakar的答案要好得多

x*z[…，None]

？@Divakar很有效，谢谢。如果有，请提供此类操作的文档，这可能是一个良好的开端。还可以查找None/np.newaxis。

x*np.dstack（（z，z，z））

也可以，但是@Divakar的答案要好得多

x = np.array([
  [[1,2,3], [4,5,6], [1,2,1]],
  [[1,2,3], [4,5,6], [3, 5, 8]]
])

z = np.array([
  [1, 0, 1],
  [0,1, 0]
])

#output
[
  [[1,2,3], [0,0,0], [1,2,1]],
  [[0,0,0], [4,5,6], [0,0,0]]
]