代码之家  ›  专栏  ›  技术社区  ›  Daniel Mahler

带广播的numpy阵列构造

  •  1
  • Daniel Mahler  · 技术社区  · 6 年前

    我希望将具有不同但广播兼容阵列的阵列堆叠在一起。给定7x5,7x1,1x5和1x1数组,我想

     a475 = mkarr([a75, a71, a15, a11])
    

    哪里 a455

     a2275 = mkarr([[a75, a71], [a15, a11]])
    

    获取2x2x7x5阵列。

    1 回复  |  直到 6 年前
        1
  •  3
  •   javidcf    6 年前

    你可以用 np.broadcast_arrays :

    a475 = np.stack(np.broadcast_arrays(a75, a71, a15, a11))