Vrátiť nepárne prvky s / bez bloku - ruby, blok, výnos

Týmto spôsobom sa snažím vrátiť nepárne prvky

def odd_elements(array)
retArr = Array.new
array.each_with_index do |item, index|
if block_given?
yield(item)
else
retArr << item
end if index % 2 != 0
end

return retArr
end

aby som mohol absolvovať obidva tieto testy

it "should yield odd elements" do
res = odd_elements([1,2,3,4,5,6])
res.should be_an_instance_of Array
res.should have(3).items
res.should include(2)
res.should include(4)
res.should include(6)
end

it "should yield" do
res = odd_elements([1,2,3,4,5,6]) {|x| x**2 }
res.should be_an_instance_of Array
res.should have(3).items
res[0].should == 4
res[1].should == 16
res[2].should == 36
end

ale v druhom zlyhávam. Zdá sa, že nerozumiem tomu, ako vyťažiť, a nedokázal som to napraviť za dve hodiny skúšaním toľkých rôznych vecí. Mohli by ste mi prosím vysvetliť, prečo to nefunguje?

odpovede:

yield vráti hodnotu, ktorú chcete pridať retArr:

def odd_elements(array)
retArr = []
array.each_with_index do |item, index|
retArr << (block_given? ? yield(item) : item) if index % 2 != 0
end
retArr
end

Ja osobne by som radšej Enumerator (ak používate Ruby 1.9.3+), trochu pomalšie, ale lepšie čitateľné a flexibilnejšie:

def odd_elements(array)
Enumerator.new do |e|
odd = false
array.each do |x|
e << x if odd
odd = ! odd
end
end
end

a = [1,2,3,4]

odd_elements(a).to_a #=> [2, 4]
odd_elements(a).map { |x| x * 2 } #=> [4, 8]

odd_elements([1,2,3,4,5,6]){|x|p x**2 } #note the p
#=> 4
#=> 16
#=> 36

Poskytuje sa v poriadku, ale výsledok nikdy neuložíte do poľa.

odd_elements([1,2,3,4,5,6]).map {|x| x**2 }

uloží výsledok do poľa.