(在开始之前,我使用的是Ruby 1.8.7,所以我不能使用花哨的东西。)
正如标题所说,我想计算列或行的平均值。 但是,我甚至无法找到从在线遍历/迭代Matrix形式的数组的方法。
让我们说你有这个
require 'mathn'
m = Matrix[[1,2,3],[4,5,6],[7,8,9]]
不知怎的,我迭代一个简单的3x3数组的方式并不适用于矩阵形式的数组(或者可能只是我的代码很奇怪)..这样做的正确方法是什么?另外,是否有一种语法可以计算矩阵的行和列平均值
答案 0 :(得分:2)
这是计算给定矩阵中特定行或列的平均值的一种方法:
require 'matrix'
m = Matrix[
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
def vector_average(matrix, vector_type, vector_index)
vector = matrix.send(vector_type, vector_index)
vector.inject(:+) / vector.size.to_f
end
# Average of first row
vector_average(m, :row, 0)
# => 2.0
# Average of second column
vector_average(m, :column, 1)
# => 5.0
希望这有帮助!
答案 1 :(得分:1)
如果您希望计算所有行平均值和/或所有列平均值,您可以执行以下操作。
require 'matrix'
def row_averages(m)
(m * Vector[*[1.0/m.column_size]*m.column_size]).to_a
end
def col_averages(m)
row_averages(m.transpose)
end
例如,
m = Matrix[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
row_averages(m)
#=> [2.0, 5.0, 8.0]
col_averages(m)
#=> [3.9999999999999996, 5.0, 6.0]