我想mapcar
在 Ruby 中实现 Lisp。
如意语法:
mul = -> (*args) { args.reduce(:*) }
mapcar(mul, [1,2,3], [4,5], [6]) would yield [24, nil, nil].
这是我能想到的解决方案:
arrs[0].zip(arrs[1], arrs[2]) => [[1, 4, 6], [2, 5, nil], [3, nil, nil]]
然后我可以:
[[1, 4, 6], [2, 5, nil], [3, nil, nil]].map do |e|
e.reduce(&mul) unless e.include?(nil)
end
=> [24, nil, nil]
但我被困在这zip
部分。如果输入为[[1], [1,2], [1,2,3], [1,2,3,4]]
,则该zip
部分需要更改为:
arrs[0].zip(arrs[1], arrs[2], arrs[3])
对于两个输入数组,我可以这样写:
def mapcar2(fn, *arrs)
return [] if arrs.empty? or arrs.include? []
arrs[0].zip(arrs[1]).map do |e|
e.reduce(&fn) unless e.include? nil
end.compact
end
但我不知道如何超越两个以上的数组:
def mapcar(fn, *arrs)
# Do not know how to abstract this
# zipped = arrs[0].zip(arrs[1], arrs[2]..., arrs[n-1])
# where n is the size of arrs
zipped.map do |e|
e.reduce(&fn) unless e.include?(nil)
end.compact
end
有人有建议吗?