如何在 repa 中堆叠数组 (Haskell)

Question

假设有两个相同长度的一维数组：

let x = fromListUnboxed (ix1 4) [1, 2, 3, 4]
let y = fromListUnboxed (ix1 4) [5, 6, 7, 8]

现在我想把这两个数组堆叠成一个二维数组，这样这些数组就形成了行。我怎样才能在 repa 中做到这一点？

基本上，我正在寻找 numpy 的等价物 row_stack:

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

注意。 x 和 y 这两个数组来自外部，即我无法从头创建二维数组。

Answer 1

正如我在最初的评论中提到的，您只需要 reshape 然后 append（都在 Data.Array.Repa.

ghci> let x' = reshape (ix2 4 1) x
ghci> let y' = reshape (ix2 4 1) y
ghci> z <- computeP $ x' `append` y' :: IO (Array U DIM2 Int)
ghci> z
AUnboxed ((Z :. 4) :. 2) [1,5,2,6,3,7,4,8]

至于漂亮的印刷，repa 不是很好（可能是因为没有很好的漂亮印刷用于更高的维度）。这是显示 z

的单行技巧

ghci> putStr $ unlines [ unwords [ show $ z ! ix2 i j  |  i<-[0..3] ] | j<-[0..1] ]
1 2 3 4
5 6 7 8

如何在 repa 中堆叠数组 (Haskell)

How to stack arrays in repa (Haskell)

arrays

haskell

repa