如何从矩阵[数组]中获得平均值

时间:2018-05-02 22:30:52

标签: ruby matrix average ruby-1.8.7

(在开始之前,我使用的是Ruby 1.8.7,所以我不能使用花哨的东西。)

正如标题所说,我想计算列或行的平均值。 但是,我甚至无法找到从在线遍历/迭代Matrix形式的数组的方法。

让我们说你有这个

require 'mathn'

m = Matrix[[1,2,3],[4,5,6],[7,8,9]]

不知怎的,我迭代一个简单的3x3数组的方式并不适用于矩阵形式的数组(或者可能只是我的代码很奇怪)..这样做的正确方法是什么?另外,是否有一种语法可以计算矩阵的行和列平均值

2 个答案:

答案 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]

请参阅MatrixVector