Skip to content
Related Articles

Related Articles

Improve Article

Find the mean vector of a Matrix

  • Difficulty Level : Basic
  • Last Updated : 13 Apr, 2021

Given a matrix of size M x N, the task is to find the Mean Vector of the given matrix. 
Examples: 
 

Input : mat[][] = {{1, 2, 3},
                   {4, 5, 6},
                   {7, 8, 9}}       
Output : Mean Vector is [4 5 6]
Mean of column 1 is (1 + 4 + 7) / 3 = 4
Mean of column 2 is (2 + 5 + 8) / 3 = 5
Mean of column 3 is (3 + 6 + 9) / 3 = 6

Input : mat[][] =  {{2, 4},
                    {6, 8}}
Output : Mean Vector is [4 6]
Mean of column 1 is (2 + 6) / 2 = 4
Mean of column 2 is (4 + 8) / 2 = 6

Approach: 
Lets take a matrix mat of dimension 5×3 representing lengths, breadths, heights of 5 objects. 
Now, the resulting mean vector will be a row vector of the following format : 
 

[mean(length) mean(breadth)  mean(height)]

Note: If we have a matrix of dimension M x N, then the resulting row vector will be having dimension 1 x N 
Now, simply calculate the mean of each column of the matrix which will give the required mean vector .
 

C++




// C++ program to find mean vector
// of given matrix
#include <bits/stdc++.h>
using namespace std;
#define rows 3
#define cols 3
 
// Function to find mean vector
void meanVector(int mat[rows][cols])
{
    cout << "[ ";
 
    // loop to traverse each column
    for (int i = 0; i < rows; i++) {
 
        // to calculate mean of each row
        double mean = 0.00;
 
        // to store sum of elements of a column
        int sum = 0;
 
        for (int j = 0; j < cols; j++)
            sum += mat[j][i];
 
        mean = sum / rows;
        cout << mean << " ";
    }
 
    cout << "]";
}
 
// Drivers code
int main()
{
 
    int mat[rows][cols] = { { 1, 2, 3 },
                            { 4, 5, 6 },
                            { 7, 8, 9 } };
 
    meanVector(mat);
 
    return 0;
}

Java




// Java program to find
// mean vector of given matrix
import java.io.*;
 
class GFG
{
static int rows = 3;
static int cols = 3;
 
// Function to
// find mean vector
static void meanVector(int mat[][])
{
    System.out.print("[ ");
 
    // loop to traverse
    // each column
    for (int i = 0; i < rows; i++)
    {
 
        // to calculate mean
        // of each row
        double mean = 0.00;
 
        // to store sum of
        // elements of a column
        int sum = 0;
 
        for (int j = 0; j < cols; j++)
            sum += mat[j][i];
 
        mean = sum / rows;
        System.out.print((int)mean + " ");
    }
 
    System.out.print("]");
}
 
// Driver code
public static void main (String[] args)
{
    int mat[][] = {{1, 2, 3},
                   {4, 5, 6},
                   {7, 8, 9}};
 
    meanVector(mat);
}
}
 
// This code is contributed
// by anuj_67.

Python3




# Python3 program to find
# mean vector of given
# matrix
rows = 3;
cols = 3;
 
# Function to
# find mean vector
def meanVector(mat):
    print("[ ", end = "");
 
    # loop to traverse
    # each column
    for i in range(rows):
         
        # to calculate
        # mean of each row
        mean = 0.00;
 
        # to store sum of
        # elements of a column
        sum = 0;
 
        for j in range(cols):
            sum = sum + mat[j][i];
 
        mean = int(sum /rows);
        print(mean, end = " ");
 
    print("]");
 
# Driver Code
mat = [[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]];
 
meanVector(mat);
 
# This code is contributed
# by mits

C#




// C# program to find
// mean vector of given matrix
using System;
 
class GFG
{
static int rows = 3;
static int cols = 3;
 
// Function to
// find mean vector
static void meanVector(int [,]mat)
{
    Console.Write("[ ");
 
    // loop to traverse
    // each column
    for (int i = 0; i < rows; i++)
    {
 
        // to calculate mean
        // of each row
        double mean = 0.00;
 
        // to store sum of
        // elements of a column
        int sum = 0;
 
        for (int j = 0; j < cols; j++)
            sum += mat[j, i];
 
        mean = sum / rows;
        Console.Write((int)mean + " ");
    }
 
    Console.Write("]");
}
 
// Driver code
public static void Main ()
{
    int[,] mat = {{1, 2, 3},
                  {4, 5, 6},
                  {7, 8, 9}};
 
    meanVector(mat);
}
}
 
// This code is contributed
// by anuj_67.

PHP




<?php
// PHP program to find mean
// vector of given matrix
$rows = 3;
$cols = 3;
 
// Function to find mean vector
function meanVector($mat)
{
    global $rows ,$cols;
    echo "[ ";
 
    // loop to traverse
    // each column
    for ($i = 0; $i < $rows; $i++)
    {
 
        // to calculate
        // mean of each row
        $mean = 0.00;
 
        // to store sum of
        // elements of a column
        $sum = 0;
 
        for ($j = 0; $j < $cols; $j++)
            $sum += $mat[$j][$i];
 
        $mean = $sum /$rows;
        echo $mean , " ";
    }
 
    echo "]";
}
 
// Driver Code
$mat = array(array(1, 2, 3),
             array(4, 5, 6),
             array(7, 8, 9));
 
meanVector($mat);
 
// This code is contributed
// by anuj_6
?>

Javascript




<script>
 
// Javascript program to find
// mean vector of given matrix
var rows = 3;
var cols = 3;
 
// Function to find mean vector
function meanVector(mat)
{
    document.write("[ ");
 
    // Loop to traverse
    // each column
    for (var i = 0; i < rows; i++)
    {
         
        // To calculate mean
        // of each row
        var mean = 0.00;
 
        // to store sum of
        // elements of a column
        var sum = 0;
 
        for (var j = 0; j < cols; j++)
            sum += mat[j][i];
 
        mean = sum / rows;
        document.write(mean + " ");
    }
 
    document.write("]");
}
 
// Driver code
var mat = [ [ 1, 2, 3 ],
            [ 4, 5, 6 ],
            [ 7, 8, 9 ] ];
 
meanVector(mat);
            
// This code is contributed by Kirti
 
</script>
Output: 
[ 4 5 6 ]

 

Time Complexity: O(rows * cols)
 

Attention reader! Don’t stop learning now. Get hold of all the important DSA concepts with the DSA Self Paced Course at a student-friendly price and become industry ready.  To complete your preparation from learning a language to DS Algo and many more,  please refer Complete Interview Preparation Course.

In case you wish to attend live classes with experts, please refer DSA Live Classes for Working Professionals and Competitive Programming Live for Students.




My Personal Notes arrow_drop_up
Recommended Articles
Page :