Root Mean Square of a Function in Python

I want to calculate root mean square of a function in Python. My function is in a simple form like y = f(x). x and y are arrays.

I tried Numpy and Scipy Docs and couldn't find anything.

2

2 Answers

I'm going to assume that you want to compute the expression given by the following pseudocode:

ms = 0
for i = 1 ... N
    ms = ms + y[i]^2
ms = ms / N
rms = sqrt(ms)

i.e. the square root of the mean of the squared values of elements of y.

In numpy, you can simply square y, take its mean and then its square root as follows:

rms = np.sqrt(np.mean(y**2))

So, for example:

>>> y = np.array([0, 0, 1, 1, 0, 1, 0, 1, 1, 1])  # Six 1's
>>> y.size
10
>>> np.mean(y**2)
0.59999999999999998
>>> np.sqrt(np.mean(y**2))
0.7745966692414834

Do clarify your question if you mean to ask something else.

4

You could use the sklearn function

from sklearn.metrics import mean_squared_error
rmse = mean_squared_error(y_actual,[0 for _ in y_actual], squared=False)

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Robert Thorne

Robert Thorne

Automotive & Future Transportation Editor

Robert Thorne covers electric vehicle innovations, autonomous driving systems, global mobility trends, and automotive engineering developments.

Share this article
Twitter Facebook Pinterest