如何复制numpy数组中最后一个维度的值？

Question

当我执行两个数组a和b的成对乘法时，由于a的维数为100 x 200 x 3，而b的维数为100 x 200，我在numpy中得到了一个错误。但是，b只包含0和1s。如何将b的最后维数重复3次才能将b转换为100x200x3数组？

这与matlab中的爬虫类似。我基本上想要三份b的最后一个维度。我试过了np.tile(b, (1, 1, 3))，但是这产生了错误的维度。

Answer 1

a * b[..., np.newaxis]

在最后给b另一个长度-1轴，广播将为您处理这个问题，而不需要实际构造一个三倍的数组。