matlab - 将多维元胞数组的生成从 Matlab 转换为 Clojure 的最佳方法是什么

Question

我正在为我的问题找出解决方案，但我感觉它不会非常有效。我有一个可变长度数组的二维单元格结构，它是在 Matlab 中以非常非功能性的方式构建的，我想将其转换为 Clojure。这是我正在尝试做的一个例子:

pre = cell(N,1);
aux = cell(N,1);
for i=1:Ne
  for j=1:D
    for k=1:length(delays{i,j})
        pre{post(i, delays{i, j}(k))}(end+1) = N*(delays{i, j}(k)-1)+i;
        aux{post(i, delays{i, j}(k))}(end+1) = N*(D-1-j)+i; % takes into account delay
    end;
  end;
end;

我目前的实现计划是使用 3 个循环，其中第一个循环是用一个由 N 个向量组成的空向量向量初始化的。每个子循环都由前一个循环初始化。我定义了一个单独的函数，它接受整体向量和子索引和值，并返回带有更新子向量的向量。

必须有比使用 3 循环/递归更聪明的方法来做到这一点。可能是一些 reduce 函数通过使用累加器来简化语法。

Answer 1

我不是 100% 确定我理解您的代码在做什么(我不知道 Matlab)，但这可能是构建多维向量的一种方法:

(defn conj-in
  "Based on clojure.core/assoc-in, but with vectors instead of maps."
  [coll [k & ks] v]
  (if ks
    (assoc coll k (conj-in (get coll k []) ks v))
    (assoc coll k v)))

(defn foo []
  (let [w 5, h 4, d 3
        indices (for [i (range w)
                      j (range h)
                      k (range d)]
                  [i j k])]
    (reduce (fn [acc [i j k :as index]]
              (conj-in acc index
                       ;; do real work here
                       (str i j k)))
            [] indices)))

user> (pprint (foo))
[[["000" "001" "002"]
  ["010" "011" "012"]
  ["020" "021" "022"]
  ["030" "031" "032"]]
 [["100" "101" "102"]
  ["110" "111" "112"]
  ["120" "121" "122"]
  ["130" "131" "132"]]
 [["200" "201" "202"]
  ["210" "211" "212"]
  ["220" "221" "222"]
  ["230" "231" "232"]]
 [["300" "301" "302"]
  ["310" "311" "312"]
  ["320" "321" "322"]
  ["330" "331" "332"]]
 [["400" "401" "402"]
  ["410" "411" "412"]
  ["420" "421" "422"]
  ["430" "431" "432"]]]

这只有在 indices 以正确的顺序(递增)时才有效，因为你不能在任何地方conj 或 assoc 到向量上除了最后一个。

我还认为使用 make-array 并通过 aset 构建数组是可以接受的。这就是 Clojure 提供访问 Java 可变数组的原因；一些算法以这种方式更加优雅，有时您需要它们来提高性能。如果你想避免泄漏副作用，你总是可以在完成后将数据转储到 Clojure 向量中。

(我不知道这个版本和其他版本哪个性能更好。)

(defn bar []
  (let [w 5, h 4, d 3
        arr (make-array String w h d)]
    (doseq [i (range w)
            j (range h)
            k (range d)]
      (aset arr i j k (str i j k)))
    (vec (map #(vec (map vec %)) arr))))  ;yikes?