Clojure列表理解-在特定情况下只运行一次内部循环

Question

有没有一种方法可以使用单个列表组件来实现以下目标，或者我需要其他方法？

(def args '([[:db/foo 1 2] [:db/bar 3 4]] {:a 1 :b 2}))

(defn myfunc []
"Want to run inner loop only once if arg is a map."
(for [arg args [cmd & params] arg] ; 'cmd & params' are meaningless if map.
(if (map? arg)
arg
(into [cmd] params)))) 

以上代码产生

=> [[:db/foo 1 2] [:db/bar 3 4] {:a 1, :b 2} {:a 1, :b 2}]

但我实际上想要

=> [[:db/foo 1 2] [:db/bar 3 4] {:a 1, :b 2}]

显然，这不是完整的函数，如果arg是向量形式，但希望让映射形式直接通过(而不重复(，我正在进行转换。

Update：我发现列表理解的根源在于集合生成器表示法，而"有条件执行内部循环"实际上是一个命令式概念，所以用一个完整的列表理解来表达这一点并不奇怪，这完全是一个fp构造。

Answer 1

这就是你想要做的吗？：从现有序列构建新序列。对于输入序列中的每个元素，都有两种可能的情况：

• 这是一个映射：在这种情况下，只需将其附加到结果序列中
• 否则，它就是一个序列，它的元素可能会被转换并附加到结果序列中

如果这就是您想要做的，则可以通过首先将输入序列的元素映射到子序列，然后将这些序列连接起来以形成结果序列来方便地表达此计算。

正如@cfrick所建议的，有一种mapcat将映射与级联相结合，要么通过将序列直接传递给mapcat

(mapcat (fn [x] (if (map? x) [x] x)) args)

或者通过使用mapcat作为换能器

(into [] (mapcat (fn [x] (if (map? x) [x] x))) args)

我认为for不适合表达这种计算。

Answer 2

在我提出解决方案之前，我对您提供的函数myfunc有几点意见：

• 文档字符串放错地方了。我知道这很奇怪，但doc字符串必须在函数的参数之前，所以在[]之前
• 我认为你对for指令的使用并不是你所期望的。您应该看看这里文档中的示例。for指令主要用于从列表构造笛卡尔乘积

我希望我能正确理解你想要实现的目标。在我的命题中，我使用通过loop指令的递归来构建一个向量，该向量将包含所有参数：

(defn myfunc "Want to run inner loop only once if arg is a map."
[& args]
(loop [arg-to-parse args
res          []]
(if (empty? arg-to-parse)
res
(let [arg              (first arg-to-parse)
new-arg-to-parse (rest  arg-to-parse)]
(if (map? arg)
(recur new-arg-to-parse (conj res arg))
(recur new-arg-to-parse (into res arg))
)))))
(myfunc [[:db/foo 1 2] [:db/bar 3 4]] {:a 1 :b 2})
;; => [[:db/foo 1 2] [:db/bar 3 4] {:a 1, :b 2}]
(myfunc {:a 1 :b 2} [[:db/foo 1 2] [:db/bar 3 4]])
;; => [{:a 1, :b 2} [:db/foo 1 2] [:db/bar 3 4]]