MATLAB to Python Conversion Arrays

Question

我在MATLAB中有一些代码，我试图转换成python。我对python知之甚少，所以这是一个挑战。

这里的MATLAB代码：

xm_row = -(Nx-1)/2.0+0.5:(Nx-1)/2.0-0.5; 
xm = xm_row(ones(Ny-1, 1), :); 
ym_col = (-(Ny-1)/2.0+0.5:(Ny-1)/2.0-0.5)'; 
ym = ym_col(:,ones(Nx-1,1)); 

这里是我试图做同样的事情在Python非常粗糙的尝试：

for x in range (L-1): 
    for y in range (L-1): 
     xm_row = x[((x-1)/2.0+0.5):((x-1)/2.0-.5)] 
     xm = xm_row[(ones(y-1,1)),:] 
     ym_column = transposey[(-(y-1)/2.0+0.5):((y-1)/2.0-.5)] 
     ym = ym_column[:,ones(x-1,1)] 

在我的Python代码，L是大小我正在循环的阵列。 当我尝试在Python运行它，我到达那里的错误：

'int' object has no attribute '__getitem__' 

在该行：

xm_row = x[((x-1)/2.0+0.5):((x-1)/2.0-.5)] 

任何帮助表示赞赏！

Answer 1

在MATLAB中，可以实现在一个简单的方式与meshgrid，像这样 -

Nx = 5; 
Ny = 7; 

xm_row = -(Nx-1)/2.0+0.5:(Nx-1)/2.0-0.5; 
ym_col = (-(Ny-1)/2.0+0.5:(Ny-1)/2.0-0.5)'; 
[xm_out,ym_out] = meshgrid(xm_row,ym_col) 

让我们比较这meshgrid版本进行验证原代码 -

>> Nx = 5; 
>> Ny = 7; 
>> xm_row = -(Nx-1)/2.0+0.5:(Nx-1)/2.0-0.5; 
>> ym_col = (-(Ny-1)/2.0+0.5:(Ny-1)/2.0-0.5)'; 
>> xm = xm_row(ones(Ny-1, 1), :) 
xm = 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
>> ym = ym_col(:,ones(Nx-1,1)) 
ym = 
     -2.5   -2.5   -2.5   -2.5 
     -1.5   -1.5   -1.5   -1.5 
     -0.5   -0.5   -0.5   -0.5 
      0.5   0.5   0.5   0.5 
      1.5   1.5   1.5   1.5 
      2.5   2.5   2.5   2.5 
>> [xm_out,ym_out] = meshgrid(xm_row,ym_col) 
xm_out = 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
     -1.5   -0.5   0.5   1.5 
ym_out = 
     -2.5   -2.5   -2.5   -2.5 
     -1.5   -1.5   -1.5   -1.5 
     -0.5   -0.5   -0.5   -0.5 
      0.5   0.5   0.5   0.5 
      1.5   1.5   1.5   1.5 
      2.5   2.5   2.5   2.5 

现在，过渡从MATLAB到Python在NumPy中有一个更简单的介质，因为它承载了许多来自MATLAB的对应用于Python环境。对于我们的情况，我们有meshgrid的NumPy的版本，这使得它只是一个直接的移植下面所列 -

import numpy as np # Import NumPy module 

Nx = 5; 
Ny = 7; 

# Use np.arange that is a colon counterpart in NumPy/Python 
xm_row = np.arange(-(Nx-1)/2.0+0.5,(Nx-1)/2.0-0.5+1) 
ym_col = np.arange(-(Ny-1)/2.0+0.5,(Ny-1)/2.0-0.5+1) 

# Use meshgrid just like in MATLAB 
xm,ym = np.meshgrid(xm_row,ym_col) 

输出 -

In [28]: xm 
Out[28]: 
array([[-1.5, -0.5, 0.5, 1.5], 
     [-1.5, -0.5, 0.5, 1.5], 
     [-1.5, -0.5, 0.5, 1.5], 
     [-1.5, -0.5, 0.5, 1.5], 
     [-1.5, -0.5, 0.5, 1.5], 
     [-1.5, -0.5, 0.5, 1.5]]) 

In [29]: ym 
Out[29]: 
array([[-2.5, -2.5, -2.5, -2.5], 
     [-1.5, -1.5, -1.5, -1.5], 
     [-0.5, -0.5, -0.5, -0.5], 
     [ 0.5, 0.5, 0.5, 0.5], 
     [ 1.5, 1.5, 1.5, 1.5], 
     [ 2.5, 2.5, 2.5, 2.5]]) 

而且，请注意，目前正在加入+1在这两种情况下np.arange的第二个参数的末尾，因为np.arange排除了创建元素范围时的第二个参数元素。作为一个例子，如果我们想创建一个范围从3元素10的，我们将需要做np.arange(3,10+1)如下 -

In [32]: np.arange(3,10+1) 
Out[32]: array([ 3, 4, 5, 6, 7, 8, 9, 10])