如何复制 numpy 数组中的行或列？

有了这个numpy array:

[[0 1 2]
 [3 4 5]
 [6 7 8]] 

例如，我如何复制第 1 行，以便得到以下内容？：

[[0 1 2]
 [3 4 5]
 [3 4 5]
 [6 7 8]] 

方法#1

一种方法是np.insert https://docs.scipy.org/doc/numpy-1.12.0/reference/generated/numpy.insert.html -

np.insert(a,2,a[1],axis=0)

对于复制列，请一起使用axis=1 -

np.insert(a,2,a[:,1],axis=1)

将其作为具有通用重复次数的函数 -

def dup_rows(a, indx, num_dups=1):
    return np.insert(a,[indx+1]*num_dups,a[indx],axis=0)

def dup_cols(a, indx, num_dups=1):
    return np.insert(a,[indx+1]*num_dups,a[:,[indx]],axis=1)

样本运行 -

In [82]: a
Out[82]: 
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

In [83]: np.insert(a,2,a[1],axis=0)
Out[83]: 
array([[0, 1, 2],
       [3, 4, 5],
       [3, 4, 5],
       [6, 7, 8]])

In [141]: np.insert(a,2,a[:,1],axis=1)
Out[141]: 
array([[0, 1, 1, 2],
       [3, 4, 4, 5],
       [6, 7, 7, 8]])

通用案例运行 -

In [255]: a
Out[255]: 
array([[19, 65, 87, 46, 85],
       [18, 45, 90, 26, 31],
       [49, 35, 34, 62, 24],
       [47, 85, 63, 91, 33],
       [54, 37, 89, 79, 50],
       [53, 54, 66, 59, 38]])

In [256]: dup_rows(a, indx=4, num_dups=3)
Out[256]: 
array([[19, 65, 87, 46, 85],
       [18, 45, 90, 26, 31],
       [49, 35, 34, 62, 24],
       [47, 85, 63, 91, 33],
       [54, 37, 89, 79, 50],
       [54, 37, 89, 79, 50],
       [54, 37, 89, 79, 50],
       [54, 37, 89, 79, 50],
       [53, 54, 66, 59, 38]])

In [253]: dup_cols(a, indx=2, num_dups=2)
Out[253]: 
array([[19, 65, 87, 87, 87, 46, 85],
       [18, 45, 90, 90, 90, 26, 31],
       [49, 35, 34, 34, 34, 62, 24],
       [47, 85, 63, 63, 63, 91, 33],
       [54, 37, 89, 89, 89, 79, 50],
       [53, 54, 66, 66, 66, 59, 38]])

方法#2

另一个与np.repeat https://docs.scipy.org/doc/numpy/reference/generated/numpy.repeat.html -

In [102]: reps = np.ones(a.shape[0],dtype=int)

In [103]: reps[1] = 2 # duplication factor

In [104]: np.repeat(a,reps,axis=0)
Out[104]: 
array([[0, 1, 2],
       [3, 4, 5],
       [3, 4, 5],
       [6, 7, 8]])