我有一个整数数组。
例如:
array = [123,321,12389]
有什么好方法可以得到它们的总和吗?
我知道
sum = 0
array.each { |a| sum+=a }
会工作。
array.sum
Array
为空时提供默认值,例如如果您想要返回一个 Money
对象而不是 Integer
,您可以执行类似 array.sum( 0.to_money( "USD" ) )
的操作。
尝试这个:
array.inject(0){|sum,x| sum + x }
See Ruby's Enumerable Documentation
(注意:需要 0
基本情况,以便 0
将返回一个空数组而不是 nil
)
array.inject(:+)
效率更高。
array.inject(:+)
似乎在 Ruby 1.8.6 中导致 麻烦 异常“LocalJumpError : no block given”可能会弹出。
array.sum
可能会给你数组值的总和。
reduce
,它是 inject
的别名(如 array.reduce( :+ )
)。
inject
而不是 reduce
。
array.reduce(0, :+)
虽然等同于 array.inject(0, :+)
,但随着 MapReduce programming models 的兴起,reduce 一词正在进入更常见的术语。
inject、reduce、fold、accumulate 和 compress 都是同义词folding functions 类。我发现您的代码库的一致性最重要,但由于各个社区倾向于使用一个词而不是另一个词,因此了解替代词仍然很有用。
为了强调 map-reduce 的措辞,这里有一个版本,它对数组中的结果更加宽容。
array.map(&:to_i).reduce(0, :+)
一些额外的相关阅读:
http://ruby-doc.org/core-1.9.3/Enumerable.html#method-i-inject
http://en.wikipedia.org/wiki/MapReduce
http://en.wikipedia.org/wiki/Fold_(higher-order_function)
reduce
告诉我更多该函数的作用,但 inject
听起来确实更酷。
reduce
和 map
作为高阶函数早于 MapReduce。灵感是相反的。在 MapReduce 的意义上,它与简单的函数式 reduce 有点不同,对不同机器的通信方式有影响。
或者(仅用于比较),如果您安装了 Rails(实际上只是 ActiveSupport):
require 'activesupport'
array.sum
require 'active_support/core_ext/enumerable.rb'
,或者需要所有的主动支持:require 'active_support/all'
。更多信息在这里:API Docs
activesupport
是一个 大量 依赖项,可以拖入项目以从 array.inject(:+)
转到 array.sum
。
.rb
后缀的 require 'active_support/core_ext/enumerable'
,因为它是隐式添加的。
对于 Ruby >=2.4.0,您可以使用 Enumerables 中的 sum
。
[1, 2, 3, 4].sum
mokeypatch 基类是危险的。如果您喜欢危险并使用旧版本的 Ruby,您可以将 #sum
添加到 Array
类:
class Array
def sum
inject(0) { |sum, x| sum + x }
end
end
Ruby 2.4.0 的新功能
您可以使用恰当命名的方法 Enumerable#sum
。它比 inject(:+)
有很多优点,但最后也有一些重要的注释需要阅读。
例子
范围
(1..100).sum
#=> 5050
数组
[1, 2, 4, 9, 2, 3].sum
#=> 21
[1.9, 6.3, 20.3, 49.2].sum
#=> 77.7
重要的提示
此方法不等同于 #inject(:+)
。例如
%w(a b c).inject(:+)
#=> "abc"
%w(a b c).sum
#=> TypeError: String can't be coerced into Integer
还,
(1..1000000000).sum
#=> 500000000500000000 (execution time: less than 1s)
(1..1000000000).inject(:+)
#=> 500000000500000000 (execution time: upwards of a minute)
有关 sum
为何如此的详细信息,请参阅 this answer。
Array
为空时提供默认值,例如如果您想要返回一个 Money
对象而不是 Integer
,您可以执行类似 array.sum( 0.to_money( "USD" ) )
的操作。
Ruby 2.4+ / Rails - array.sum
即 [1, 2, 3].sum # => 6
Ruby pre 2.4 - array.inject(:+)
或 array.reduce(:+)
*注意:#sum
方法是 2.4 对 enumerable
的新增功能,因此您现在可以在纯 ruby 中使用 array.sum
,而不仅仅是 Rails。
只是为了多样性,如果您的数组不是数字数组,而是具有数字属性(例如数量)的对象数组,您也可以这样做:
array.inject(0){|sum,x| sum + x.amount}
array.map(&:amount).inject(0, :+)
。查看其他答案。
map
然后 inject
需要您循环遍历数组两次:一次是创建一个新数组,另一次是对成员求和。这种方法略显冗长,但也更有效。
ruby 1.8.7 方式如下:
array.inject(0, &:+)
Ruby 2.4.0 已发布,它有一个 Enumerable#sum 方法。所以你可以做
array.sum
文档中的示例:
{ 1 => 10, 2 => 20 }.sum {|k, v| k * v } #=> 50
(1..10).sum #=> 55
(1..10).sum {|v| v * 2 } #=> 110
对于具有 nil 值的数组,我们可以进行压缩,然后注入总和 ex-
a = [1,2,3,4,5,12,23.45,nil,23,nil]
puts a.compact.inject(:+)
还允许 [1,2].sum{|x| x * 2 } == 6
:
# http://madeofcode.com/posts/74-ruby-core-extension-array-sum
class Array
def sum(method = nil, &block)
if block_given?
raise ArgumentError, "You cannot pass a block and a method!" if method
inject(0) { |sum, i| sum + yield(i) }
elsif method
inject(0) { |sum, i| sum + i.send(method) }
else
inject(0) { |sum, i| sum + i }
end
end
end
方法一:
[1] pry(main)> [1,2,3,4].sum
=> 10
[2] pry(main)> [].sum
=> 0
[3] pry(main)> [1,2,3,5,nil].sum
TypeError: nil can't be coerced into Integer
方法二:
[24] pry(main)> [].inject(:+)
=> nil
[25] pry(main)> [].inject(0, :+)
=> 0
[4] pry(main)> [1,2,3,4,5].inject(0, :+)
=> 15
[5] pry(main)> [1,2,3,4,nil].inject(0, :+)
TypeError: nil can't be coerced into Integer
from (pry):5:in `+'
方法三:
[6] pry(main)> [1,2,3].reduce(:+)
=> 6
[9] pry(main)> [].reduce(:+)
=> nil
[7] pry(main)> [1,2,nil].reduce(:+)
TypeError: nil can't be coerced into Integer
from (pry):7:in `+'
方法4:当Array包含nil和空值时,默认情况下,如果你使用上述任何函数reduce,sum,inject一切都会通过
TypeError:nil 不能被强制转换为 Integer
你可以通过以下方式克服这个问题,
[16] pry(main)> sum = 0
=> 0
[17] pry(main)> [1,2,3,4,nil, ''].each{|a| sum+= a.to_i }
=> [1, 2, 3, 4, nil, ""]
[18] pry(main)> sum
=> 10
方法6:评估
评估字符串中的 Ruby 表达式。
[26] pry(main)> a = [1,3,4,5]
=> [1, 3, 4, 5]
[27] pry(main)> eval a.join '+'
=> 13
[30] pry(main)> a = [1,3,4,5, nil]
=> [1, 3, 4, 5, nil]
[31] pry(main)> eval a.join '+'
SyntaxError: (eval):1: syntax error, unexpected end-of-input
1+3+4+5+
如果你觉得高尔夫,你可以做
eval [123,321,12389]*?+
这将创建一个字符串“123+321+12389”,然后使用函数 eval 进行求和。这仅用于打高尔夫球的目的,您不应在正确的代码中使用它。
你也可以用简单的方法做到这一点
def sum(numbers)
return 0 if numbers.length < 1
result = 0
numbers.each { |num| result += num }
result
end
inject
或 sum
。
您可以使用 .map 和 .sum ,例如:
array.map { |e| e }.sum
array.map(&:price).inject(0, :+)
更安全一些。它确保如果你有一个空列表,你会得到 0 而不是 nil。array.inject(0) { |sum, product| sum += product.price }