How to make the individual media of each element in an array

Question

Navigation

#1 by (3 votes)

-3

Hello, I have the following array:

matriz = np.array([
    [5, 5, 5],
    [5, 5, 5],
    [10, 10, 10]
])

I need the result with an array of their respective means, like this:

[6.66, 6.66, 6.66]

Thankful

python numpy

asked by anonymous 28.08.2018 / 21:25

1 answer

Word reserved as variable name Redirect to Tab specifies [closed]

score 3 · Accepted Answer

Using the numpy.average function you can calculate the average of the array values for a given axis. Ex:

matriz = np.array([
    [5, 5, 5],
    [5, 5, 5],
    [10, 10, 10]
])

#media para o eixo Y
media = np.average(matriz, axis=0)

Output from variable media :

[6.66666667 6.66666667 6.66666667]

At documentation you can see better about the parameters.