python - 使用 numpy.ones 作为数组的索引

Question

我想将 matlab 代码翻译成 python 代码。 matlab 代码等效于以下玩具示例:

a = [1 2 3; 4 5 6; 7 8 9]
b = a(:, ones(1,3))

返回

a =

 1     2     3
 4     5     6
 7     8     9

b =

 1     1     1
 4     4     4
 7     7     7

我试着这样翻译:

from numpy import array
from numpy import ones

a = array([ [1,2,3], [4,5,6], [7,8,9] ])
b = a[:][ones((1,3))]

但它返回以下错误消息:

Traceback (most recent call last): File "example_slice.py", line 6, in b =a[:, ones((1,3))] IndexError: arrays used as indices must be of integer (or boolean) type

编辑:在这种特殊情况下，也许应该用零代替，但这不是这里的问题。该题涉及将包含相同索引的列表多次提供给数组 a 以获得与使用 Matlab 计算的数组相同的数组 b 的问题。

Answer 1

MATLAB 代码也可以写成(更地道也更清晰):

b = repmat(a(:,1),1,3);

In NumPy you'd write :

b = np.tile(a[:,None,0],(1,3))

(注意保留提取矢量的方向所需的无)。