向量化numpy数组扩展

向量化numpy数组扩展

似乎您想在

input_array

包含扩展元素的数组和数组之间进行元素 *** 作。对于这些，您可以使用

broadcasting

。

对于第一个示例，似乎您正在执行

elementwise multiplication

-

In [424]: input_array = np.array([1, 2, 3, 4])     ...: extend_array = np.array([0, 1, 1, 0])     ...:In [425]: (input_array[:,None] * extend_array).ravel()Out[425]: array([0, 1, 1, 0, 0, 2, 2, 0, 0, 3, 3, 0, 0, 4, 4, 0])

对于第二个示例，看来您正在执行

elementwise addition

-

In [422]: input_array = np.array([1, 2, 3, 4])     ...: extend_array = np.array([-0.2, -0.2, 0.2, 0.2])     ...:In [423]: (input_array[:,None] + extend_array).ravel()Out[423]: array([ 0.8,  0.8,  1.2,  1.2,  1.8,  1.8,  2.2,  2.2,  2.8,  2.8,  3.2,        3.2,  3.8,  3.8,  4.2,  4.2])