如何在Ruby中交错不同长度的数组

发布时间：2020-12-17 04:26:19 所属栏目：百科来源：网络整理

导读：如果我想在 Ruby中交错一组数组,并且每个数组的长度相同,我们可以这样做： a.zip(b).zip(c).flatten 但是,如果阵列的大小不同,我们如何解决这个问题呢？我们可以这样做： def interleave(*args) raise 'No arrays to interleave' if args.empty? max_length

如果我想在 Ruby中交错一组数组,并且每个数组的长度相同,我们可以这样做：

a.zip(b).zip(c).flatten

但是,如果阵列的大小不同,我们如何解决这个问题呢？

我们可以这样做：

def interleave(*args)
  raise 'No arrays to interleave' if args.empty?
  max_length = args.inject(0) { |length,elem| length = [length,elem.length].max }
  output = Array.new
  for i in 0...max_length
    args.each { |elem|
      output << elem[i] if i < elem.length
    }
  end
  return output
end

但有没有更好的’Ruby’方式,也许使用zip或转置或其他一些？

解决方法

如果源数组中没有nil,则只需要使用nils扩展第一个数组,zip将自动使用nil填充其他数组.这也意味着你可以使用compact来清理额外的条目,这有望比显式循环更有效

def interleave(a,*args)
    max_length = args.map(&:size).max
    padding = [nil]*[max_length-a.size,0].max
    (a+padding).zip(*args).flatten.compact
end

这是一个稍微复杂的版本,如果数组包含nil,则可以使用

def interleave(*args)
    max_length = args.map(&:size).max
    pad = Object.new()
    args = args.map{|a| a.dup.fill(pad,(a.size...max_length))}
    ([pad]*max_length).zip(*args).flatten-[pad]
end

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!