今天要講的是迭代器(iterator)。什麼是迭代器呢?它的好處就是會一個一個地傳回集合裡的元素,讓我們可以利用迭代方法做重複的事。
在Ruby裡的collection
集合裡有Array
陣列和Hash
雜湊。
今天要介紹三種用在collection集合
的迭代器分別叫each
,map
和collect
,這也是常見的Ruby面試考題呢!
Day14 each, map 和 collect 比較? What's the difference between each, map and collect?
我們來使用這三種迭代器,在Array和Hash兩種集合中各舉幾個生活化的例子,相信大家就比較容易暸解囉!
最近我在進行旅行存錢計劃。我有三個銀行帳戶,NAB
, CAN
, 和WESTPAC
,開戶金額分別為100,200,300。所以我寫一個陣列集合放入初始金額[100,200,300]。
假設我打算開始從本週開始在每個帳戶存入50元(+50),可以在陣列後加上.each
方法:
[100,200,300].each {|n| puts n+50}
結果列出各個帳戶的金額(注意:puts
寫在block
大括號裡):
150
250
350
經過.each
方法作用之後,只會分別印出同一陣列中各個元素的值,不會產生新陣列。
以上案例,換成.collect
試試看:(注意!puts寫在前面)
puts [100,200,300].collect {|n| n+50}
.collect
會幫我們把結果放入新的陣列。結果印出:
[150, 250, 350]
同樣的,.map
方法也會幫我們產生新的陣列。
puts [100,200,300].map {|n| n+50}
結果印出:
[150, 250, 350]
那.collect
與.map
又有什麼不同?以及分別用在什麼情況呢?這時候就要翻查Ruby手冊裡,關於.collect和.map的介紹了:
collect { |item| block } → new_ary
Invokes the given block once for each element of self. Creates a new array containing the values returned by the block.
hmm...好像看不出有什麼差異呢!
map { |obj| block } → array
Returns a new array with the results of running block once for every element in enum.
近一步查詢stackoverflow,果然有其他網友問過類似的問題。我們可以把.map
當作是.collect
的別名 (map is an alias for collect),實務上比較常使用.map
喔!
現在把我的銀行帳戶陣列存進account
變數裡,再用.inspect
檢查陣列的值:
account = [100, 200, 300]
account.map {|n| n+50}
p account.inspect
結果印出
"[100, 200, 300]"
如果在.map
後加上驚嘆號.map!
呢?
account.map! {|n| n+50}
p account.inspect
存進去原本的陣列了。錢錢變多了!開心~~(加上!
驚嘆號的方法,通常代表小心!注意!
的意思)
在這裡的.map
方法會讓原本的物件陣列
被改變呢!
"[150, 250, 350]"
Hash
雜湊是一對key
與value
的集合。在剛剛的銀行帳戶例子裡,我們可以把銀行名稱
當作索引
,存款數目
當作值
:
account = {"NAB" => 100, "CAN" => 200, "WEST" => 300}
利用雜湊來展現目前的資訊(帳戶名稱、存款金額),這樣就可讀性更加清楚了。
現在我想要計算三個帳戶加總共有多少錢,以.each
的方式可寫為:
mymoney = 0
account.each {|bankname, saving| mymoney += saving} #把索引和值列出
print "My Money: $ " + mymoney.to_s
或是
mymoney = 0 #設定初始值
account.each{|bank| mymoney += bank[1]} #依序加總bank集合裡第二個元素bank[1]
print "My Money: $ " + mymoney.to_s
結果都會印出:
My Money: $ 600
在Hash裡,把.each
換成.map
或是.collect
:
mymoney = 0 #設定初始值
account.collect{|bank| mymoney += bank[1]} #依序加總bank集合裡第二個元素bank[1]
print "My Money: $ " + mymoney.to_s
結果都是一樣的:
My Money: $ 600
現在要進階到一個較為複雜的例子:hash
裡包含索引
和值
兩部份,那我們可不可以把陣列
當作一種值
,放在裡面呢?
當然可以!
假設我的NAB銀行下有2個子帳戶,CAN銀行下有3個子帳戶,分別放入這些資產:
hash = { "NAB" => ["Cash", "Gold"], "CAN" => ["Bitcoin", "Litecoin", "Ethereum"] }
利用hash.map
會產生一個新的陣列:(進一步了解說明,看stackoverflow的說明)
p hash.map {|n| n}
結果顯示:
[["NAB", ["Cash", "Gold"]], ["CAN", ["Bitcoin", "Litecoin", "Ethereum"]]] #我有好多帳戶!NAB下有2個,CAN下有3個
我想分別提取出銀行:帳戶名稱
的這一對資訊,並且用逗號.join(", ")
隔開。
為了程式可讀性,hash
的索引
命名為bank
(銀行名),值
為account_arry
(放了不同數目的子帳戶陣列)。在走account_arry.each
展開陣列迭代器時,每在集合裡走完一個元素,就印出#{bank}: #{sub_account}
p hash.map {
|bank, account_arry| account_arry.each{
|sub_account| "#{bank}: #{sub_account}"}
}.join(", ")
結果僅印出:
"Cash, Gold, Bitcoin, Litecoin, Ethereum"
奇怪,這不是我要的結果呀!我很希望帳戶前面能顯示出銀行名稱呢!
這是因為剛剛說過,arry.each
會回傳陣列本身,在這個例子裡,分別回傳的是["Cash", "Gold"]
和["Bitcoin", "Litecoin", "Ethereum"]
改成.map
試試看:
p hash.map {
|bank, account_arry| account_arry.map{
|sub_account| "#{bank}: #{sub_account}"}
}.join(", ")
結果顯示為:
"NAB: Cash, NAB: Gold, CAN: Bitcoin, CAN: Litecoin, CAN: Ethereum"
這是因為account_arry.map
自動幫我們產生新的陣列,放進bank
與對應的sub_account
並回傳。
最後放個小小的比一比表格作為總結:
each | map / collect |
---|---|
Array方法 | Enumerable(列舉)方法 |
回傳Array本身 | 產生新的Array並回傳 |
祝福大家能順利collect不同的資產,不管是有形的財富、還是無形的知識,最後都可以達成錢多多的心願喔!:D
===
Ref: