Clojure列表理解-在某些情况下仅运行一次内部循环？

问题描述

有没有一种方法可以使用单个列表组合来实现以下目标？还是需要其他方法？

(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为矢量形式，但我想进行变换，但想让映射形式直接通过（不重复）。

更新：我发现list comprehension的根源是set builder notation，而“有条件地执行内部循环”实际上是一个必要的概念，因此并不是令人惊讶的是，使用 单个 列表理解很难表达这一点，这完全是fp构造。

解决方法

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

• 这是一张地图：在这种情况下，只需将其附加到结果序列中
• 否则，它是一个序列，可能应转换其元素并将其添加到结果序列中

如果这是您要执行的操作，可以通过先将输入序列的元素映射到子序列，然后将这些序列连接起来以形成结果序列来方便地表示此计算。

如@cfrick所建议，mapcat通过将序列直接传递给mapcat

来将映射与串联相结合。

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

或使用mapcat作为传感器

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

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

,

在我提出解决方案之前，我对您提供的功能myfunc进行一些说明：

• 文档字符串放错了位置。我知道这很奇怪，但是doc字符串必须在函数的参数之前，所以在[]之前，
• 您对for指令的使用不是您所期望的。您应该看一下the documentation here中的示例。主要是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]]