球拍中的 Varadic 拉链

Question

我在 Racket 中看到了otheranswers 的压缩功能，但它们首先不太正确（zip 应该只压缩到提供的最短序列，以便您可以使用无限流进行压缩) 并且最重要的是不是可变的，因此您一次只能压缩两个流。

我已经想通了

 (define (zip a-sequence b-sequence) (for/stream ([a a-sequence]
                                                  [b b-sequence])
                                       (list a b)))

确实可以正常工作

 (stream->list (zip '(a b c) (in-naturals)))
  => '((a 0) (b 1) (c 2))

但不是可变的。我知道我可以使用define (zip . sequences) 将其定义为可变参数，但我不知道如何构建for/stream 表单。

这必须是一个宏才可行吗？

Answer 1

这对你有用吗？

#lang racket

(define (my-zip . xs)
  (match xs
    [(list x) (for/stream ([e x]) (list e))]
    [(list x xs ...)
     (for/stream ([e x] [e* (apply my-zip xs)])
       (cons e e*))]))

(stream->list
 (my-zip (in-naturals) '(a b c) '(1 2 3 4 5 6)))
;;=> '((0 a 1) (1 b 2) (2 c 3))

Answer 2

zip 的常见实现是：

(require data/collection) ; for a `map` function that works with streams

(define (zip . xs)
  (apply map list xs))

...如果您更喜欢point-free style，这可以很简单：

(require racket/function)
(define zip (curry map list))

这些确实有限制，即它们要求所有输入序列具有相同的长度（包括无限长），但由于它们是惰性计算的，因此它们在任何情况下都可以工作，直到其中一个序列用完，此时会出现错误将被提高。

(zip (cycle '(a b c))
     (naturals)
     (cycle '(1 2 3))) ; => '((a 0 1) (b 1 2) (c 2 3) (a 3 1) ...

[已更新答案以反映 cmets]