首页 > 解决方案 > 有没有办法比较两个数组并返回具有相同索引的公共元素的新数组?

问题描述

例如

a = [2,3,1,1]
b = [2,7,4,2]

--> c = [2]

我的解决方案是:

c = b.select do
    |em| b.index(em) == a.index(em)
    end

但是,如果我将它应用于给定的示例,它会返回

c = [2,2]

标签: arraysruby

解决方案


由于您想按元素比较数组,zip因此这里将是一个很好的选择。

a.zip(b) # => [[2, 2], [3, 7], [1, 4], [1, 2]]

a.zip(b).select {|a1, b1| a1 == b1}.map(&:first) # [2]

# or in ruby 2.7+
a.zip(b).filter_map {|a1, b1| a1 == b1 && a1} # [2]

推荐阅读