所以我读过这个理论,现在试图在Haskell中解析一个文件 - 但是没有得到任何结论。这太奇怪了......
以下是我的输入文件的外观:
m n
k1, k2...
a11, ...., an
a21,.... a22
...
am1... amn
其中m,n只是整数,K = [k1, k2...]
是整数列表,a11..amn
是“矩阵”(列表列表):A=[[a11,...a1n], ... [am1... amn]]
这是我的快速python版本:
def parse(filename):
"""
Input of the form:
m n
k1, k2...
a11, ...., an
a21,.... a22
...
am1... amn
"""
f = open(filename)
(m,n) = f.readline().split()
m = int(m)
n = int(n)
K = [int(k) for k in f.readline().split()]
# Matrix - list of lists
A = []
for i in range(m):
row = [float(el) for el in f.readline().split()]
A.append(row)
return (m, n, K, A)
以下是我在Haskell中的表现(不是非常):
import System.Environment
import Data.List
main = do
(fname:_) <- getArgs
putStrLn fname --since putStrLn goes to IO ()monad we can't just apply it
parsed <- parse fname
putStrLn parsed
parse fname = do
contents <- readFile fname
-- ,,,missing stuff... ??? how can I get first "element" and match on it?
return contents
我对monads(以及陷阱我的上下文!)和do语句感到困惑。我真的想写这样的东西,但我知道这是错的:
firstLine <- contents.head
(m,n) <- map read (words firstLine)
因为内容不是列表 - 而是monad。
next 步骤的任何帮助都会很棒。
所以我只能discovered你能做到:
liftM lines . readFile
从文件中获取行列表。但是,该示例仍然仅转换整个ENTIRE文件,并且不仅使用第一行或第二行......
答案 0 :(得分:10)
非常简单的版本可能是:
import Control.Monad (liftM)
-- this operates purely on list of strings
-- and also will fail horribly when passed something that doesn't
-- match the pattern
parse_lines :: [String] -> (Int, Int, [Int], [[Int]])
parse_lines (mn_line : ks_line : matrix_lines) = (m, n, ks, matrix)
where [m, n] = read_ints mn_line
ks = read_ints ks_line
matrix = parse_matrix matrix_lines
-- this here is to loop through remaining lines to form a matrix
parse_matrix :: [String] -> [[Int]]
parse_matrix lines = parse_matrix' lines []
where parse_matrix' [] acc = reverse acc
parse_matrix' (l : ls) acc = parse_matrix' ls $ (read_ints l) : acc
-- this here is to give proper signature for read
read_ints :: String -> [Int]
read_ints = map read . words
-- this reads the file contents and lifts the result into IO
parse_file :: FilePath -> IO (Int, Int, [Int], [[Int]])
parse_file filename = do
file_lines <- (liftM lines . readFile) filename
return $ parse_lines file_lines
您可能希望查看Parsec以获得更好的解析,并提供更好的错误处理。
*Main Control.Monad> parse_file "test.txt"
(3,3,[1,2,3],[[1,2,3],[4,5,6],[7,8,9]])
答案 1 :(得分:10)
import Control.Monad (replicateM)
-- Read space seperated words on a line from stdin
readMany :: Read a => IO [a]
readMany = fmap (map read . words) getLine
parse :: IO (Int, Int, [Int], [[Int]])
parse = do
[m, n] <- readMany
ks <- readMany
xss <- replicateM m readMany
return (m, n, ks, xss)
我们试一试:
*Main> parse
2 2
123 321
1 2
3 4
(2,2,[123,321],[[1,2],[3,4]])
虽然我提供的代码非常具有表现力。也就是说,你可以用很少的代码快速完成工作,它有一些不好的属性。虽然我认为如果你还在学习haskell并且还没有开始使用解析器库。这是要走的路。
我的解决方案的两个不良属性:
IO
中,没有任何内容可以孤立测试[m, n]
中看到模式匹配非常激进。如果我们在输入文件的第一行有3个元素会怎样?答案 2 :(得分:9)
liftM
不是魔法!你会认为将一个函数f
提升到一个monad中会有一些神秘的事情,但它实际上只是定义为:
liftM f x = do
y <- x
return (f y)
我们实际上可以使用liftM
来做您想做的事,即:
[m,n] <- liftM (map read . words . head . lines) (readFile fname)
但你要找的是let语句:
parseLine = map read . words
parse fname = do
(x:y:xs) <- liftM lines (readFile fname)
let [m,n] = parseLine x
let ks = parseLine y
let matrix = map parseLine xs
return (m,n,ks,matrix)
正如你所看到的,我们可以使用let来表示变量赋值而不是monadic计算。事实上,当我们去做符号时,你只需要让语句表达:
parse fname =
liftM lines (readFile fname) >>= (\(x:y:xs) ->
let [m,n] = parseLine x
ks = parseLine y
matrix = map parseLine xs
in return matrix )
答案 3 :(得分:8)
使用解析库的解决方案
由于您可能会有很多人使用代码将Int
的字符串解析为[[Int]]
(map (map read . words) . lines $ contents
),我会跳过它并引入其中一个解析库。如果您要为实际工作执行此任务,您可能会使用这样的库来解析ByteString
(而不是String
,这意味着您的IO会将所有内容读入单个字符的链接列表中。 / p>
import System.Environment
import Control.Monad
import Data.Attoparsec.ByteString.Char8
import qualified Data.ByteString as B
首先,我导入了Attoparsec和bytestring库。您可以在hackage上查看这些库及其文档,并使用cabal
工具进行安装。
main = do
(fname:_) <- getArgs
putStrLn fname
parsed <- parseX fname
print parsed
main
基本没有变化。
parseX :: FilePath -> IO (Int, Int, [Int], [[Int]])
parseX fname = do
bs <- B.readFile fname
let res = parseOnly parseDrozzy bs
-- We spew the error messages right here
either (error . show) return res
parseX
(从parse重命名以避免名称冲突)使用bytestring库的readfile,它以连续的字节读取打包的文件,而不是链接列表的单元格。解析后,如果解析器返回Right result
,则使用一点简写来返回结果;如果解析器返回值Left someErrorMessage
,则使用一点错误来打印错误。
-- Helper functions, more basic than you might think, but lets ignore it
sint = skipSpace >> int
int = liftM floor number
parseDrozzy :: Parser (Int, Int, [Int], [[Int]])
parseDrozzy = do
m <- sint
n <- sint
skipSpace
ks <- manyTill sint endOfLine
arr <- count m (count n sint)
return (m,n,ks,arr)
然后真正的工作发生在parseDrozzy
。我们使用上面的帮助器获取m
和n
Int
值。在大多数Haskell解析库中,我们必须显式处理空格 - 所以我在n
之后跳过换行符到达ks
。 ks
只是下一个换行符之前的所有int值。现在我们实际上可以使用先前指定的行数和列数来获取数组。
从技术上讲,最后一位arr <- count m (count n sint)
不符合您的格式。它会抓取n
整数,即使这意味着要进入下一行。我们可以使用count m (manyTill sint endOfLine)
复制Python的行为(不验证行中值的数量),或者我们可以更明确地检查行的每一行,如果我们缺少元素,则返回错误。
从列表到矩阵
列表列表不是二维数组 - 空间和性能特征完全不同。让我们使用Data.Array.Repa(import Data.Array.Repa
)将我们的列表打包成一个真实的矩阵。这将允许我们有效地访问数组的元素以及对整个矩阵执行操作,可选地在所有可用的CPU之间分配工作。
Repa使用稍微奇怪的语法定义数组的尺寸。如果您的行和列长度位于变量m
和n
中,则Z :. n :. m
与C声明int arr[m][n]
非常相似。对于一维示例ks
,我们有:
fromList (Z :. (length ks)) ks
这会将我们的类型从[Int]
更改为Array DIM1 Int
。
对于二维数组,我们有:
let matrix = fromList (Z :. m :. n) (concat arr)
并将我们的类型从[[Int]]
更改为Array DIM2 Int
。
所以你有它。使用面向生产的库将文件格式解析为高效的Haskell数据结构。
答案 4 :(得分:4)
这样简单的事情怎么样?
parse :: String -> (Int, Int, [Int], [[Int]])
parse stuff = (m, n, ks, xss)
where (line1:line2:rest) = lines stuff
readMany = map read . words
(m:n:_) = readMany line1
ks = readMany line2
xss = take m $ map (take n . readMany) rest
main :: IO ()
main = do
stuff <- getContents
let (m, n, ks, xss) = parse stuff
print m
print n
print ks
print xss