时间戳

首页 > 解决方案 > 将 MATLAB 数组转换为 Numpy 数组

python - 将 MATLAB 数组转换为 Numpy 数组

问题描述

我正在尝试将我的 MATLAB 代码转换为 Python。

在 MATLAB 中，我有OP哪个是300x300 double数组，mask哪个是300x300 logical数组。

t1 = mask(:)等于90000x1 logical数组。

t2 = OP(mask(:))equals的输出怎么可能是一个57664x1 double数组？

这是我的 MATLAB 代码：

OP=repmat(Ph,size(image,1),1).*repmat(Pv,1,size(image,2));
t1 = mask(:)
t2 = OP(mask(:))
data=sort(OP(mask(:)),'descend');

此外，在 Python 中，我使用 Numpy 来实现我的 MATLAB 代码，但OP[mask]MATLAB 转换的代码OP(mask(:))是90000x1x300 ndarray. 我不知道如何解决它。

这是我的python代码：

OP = np.matlib.repmat(Ph, image.shape[0], 1) * np.matlib.repmat(Pv, 1, image.shape[1])
t2 = OP[mask]
data = -np.sort(-OP[mask], axis=0)

我知道 t1，OP并且mask与 MATLAB 中的类似变量具有相同的大小。

如果有人可以帮助我，我将不胜感激。

标签： pythonarraysmatlabnumpy

解决方案

在 Octave 会话中：

>> OP = reshape(1:16,4,4);
>> OP
OP =

    1    5    9   13
    2    6   10   14
    3    7   11   15
    4    8   12   16

>> mask = logical([1,0,0,1;0,1,1,1;0,0,0,0;1,0,1,0])
mask =

  1  0  0  1
  0  1  1  1
  0  0  0  0
  1  0  1  0

>> OP(mask)
ans =

    1
    4
    6
   10
   12
   13
   14

ravel 没有任何区别：

>> OP(mask(:))
ans =

    1
    4
    6
   10
   12
   13
   14

在一个ipython/numpy会话中：

In [368]: OP = np.arange(1,17).reshape(4,4)                                                            
In [369]: OP                                                                                           
Out[369]: 
array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12],
       [13, 14, 15, 16]])
In [370]: mask = np.array([[1,0,0,1],[0,1,1,1],[0,0,0,0],[1,0,1,0]]).astype(bool)                      
In [371]: mask                                                                                         
Out[371]: 
array([[ True, False, False,  True],
       [False,  True,  True,  True],
       [False, False, False, False],
       [ True, False,  True, False]])
In [372]: OP[mask]                                                                                     
Out[372]: array([ 1,  4,  6,  7,  8, 13, 15])

推荐阅读