clojure读取大文本文件并计算出现次数

时间:2013-06-23 01:03:10

标签: clojure

我正在尝试读取大型文本文件并计算特定错误的发生次数。 例如,对于以下示例文本

something
bla
error123
foo
test
error123
line
junk
error55
more
stuff

我想最终(虽然我在考虑地图,但并不关心什么是数据结构)

error123 - 2
error55 - 1

这是我到目前为止所尝试的内容

(require '[clojure.java.io :as io])

(defn find-error [line]
  (if (re-find #"error" line)    
       line))


(defn read-big-file [func, filename]
 (with-open [rdr (io/reader filename)]
   (doall (map func (line-seq rdr)))))  

像这样称呼它

 (read-big-file find-error "sample.txt")

返回:

(nil nil "error123" nil nil "error123" nil nil "error55" nil nil)

接下来,我尝试删除nil值并将项目分组

(group-by identity (remove #(= nil %) (read-big-file find-error "sample.txt")))

返回

{"error123" ["error123" "error123"], "error55" ["error55"]}

这接近所需的输出,尽管它可能效率不高。我现在该如何获得计数?此外,作为一个新的clojure和函数式编程的人,我将不胜感激任何关于如何改进它的建议。 谢谢!

1 个答案:

答案 0 :(得分:7)

我认为您可能正在寻找频率功能:

user=> (doc frequencies)
-------------------------
clojure.core/frequencies
([coll])
  Returns a map from distinct items in coll to the number of times
  they appear.
nil

所以,这应该给你你想要的东西:

(frequencies (remove nil? (read-big-file find-error "sample.txt")))
;;=> {"error123" 2, "error55" 1}

但是,如果您的文本文件非常大,我建议您在line-seq内联中执行此操作,以确保不会耗尽内存。这样您也可以使用filter而不是mapremove

(defn count-lines [pred, filename]
  (with-open [rdr (io/reader filename)]
    (frequencies (filter pred (line-seq rdr)))))

(defn is-error-line? [line]
  (re-find #"error" line))

(count-lines is-error-line? "sample.txt")
;; => {"error123" 2, "error55" 1}