I am trying to rewrite matlab code in Python

See more:

Python

MatLab

Hi, I have a code in matlab but I am trying to get the same code code in python. Here is my matlab code,

function [j,s] = J(x,Dt)
    for n = 0:1:sqrt(length(x))
    	j(n+1) = mean((x(n+1:end)-x(1:end-n)).^2);
    end
    s = Dt*[0:1:length(j)-1];

Can anyone help me to convert it in python where x shape is (10000,1,1) and Dt shape is (10000,)?

What I have tried:

I can create functions, argument and returns but cannot find exact way to write the statement inside for loop for a multidimensional arrays.