在haskell中打印2d数组

时间:2013-09-09 04:04:07

标签: arrays haskell printing ppm

我正在乱搞以这种简单的图像格式(PPM)打印随机的2d Ints数组

0 1 0 0 1
1 0 0 1 0
0 1 1 0 0
0 1 1 1 0
0 0 0 1 0

下面是我的代码,它可以工作,但对于这样一个简单的操作来说似乎太多了代码。有更好的方法吗?

import System.Random
import Data.Array.IArray

main = do
  g <- newStdGen
  let img = listArray ((0,0),(5,5)) ( myRands g ) :: Array (Int,Int) Int
  putStrLn $ printArray img

myRands :: RandomGen g => g -> [Int]
myRands g = randomRs (0,1) g

make2dRange :: Int -> Int -> Int -> Int -> [(Int,Int)]
make2dRange x1 y1 x2 y2 = range ((x1,y1),(x2,y2))

printArray :: ( Array (Int,Int) Int ) -> String
printArray arr = unlines rows
  where rows = map (unwords . map (show . (!) arr)) rowIndices 
        rowIndices = map ( \y -> make2dRange 0 y 5 y ) [0..5]

除了显而易见的,当然(例如,数组维度是硬编码的)

2 个答案:

答案 0 :(得分:7)

我会使用列表推导来操纵索引。

printArray arr =
  unlines [unwords [show (arr ! (x, y)) | x <- [0..5]] | y <- [0..5]]

答案 1 :(得分:0)

per:http://lambda.haskell.org/platform/doc/current/packages/split-0.2.2/doc/html/Data-List-Split.html#v%3achunksOf

  

chunksOf :: Int - &gt; [e] - &gt; [E]]

     

chunksOf n将列表拆分为长度为n的块。如果n不均匀地划分列表的长度,则最后一块将更短。如果n <= 0,则chunksOf n l返回无限列表的空列表。

使用chunksOf我的实现将是..

printArray arr = mapM_ (putStrLn . unwords) $ map (map show) $ chunksOf 5 arr

用法:

Prelude System.Random Data.List Data.List.Split> printArray [0..25]
0 1 2 3 4
5 6 7 8 9
10 11 12 13 14
15 16 17 18 19
20 21 22 23 24
25