如何在矩阵(python)中的所需位置插入额外的行和列

发布于04月12日

我运行了一个循环，从中获得了(4×4)数组，现在我想在这些数组中的某些位置插入零，并将它们转换成(6×6)array.

我想在第三行和第六行，以及第三列和第六列中添加零

a_obtained = np.array( [[1,     2,   3,    4   ],
                        [5,     6,   7,    8   ],
                        [9,    10,   11,  12   ],
                        [13,   14,   15,   16  ]])

我试着通过np.维斯塔克，np.hstack，但我无法在指定位置放置零.

a_desired = result = np.array([[1,     2,  0,   3,    4,   0 ],
                               [5,     6,  0,   7,    8,   0 ],
                               [0,     0,  0,   0,    0,   0 ],
                               [9,    10,  0,   11,  12,   0 ],
                               [13,   14,  0,   15,   6,   0 ],
                               [0,     0,  0,   0,    0,   0 ]])

推荐答案

如果我们有所需的列索引，可以按np.insert进行:

desired_cols_ids = np.array([2, 4], dtype=np.int64)

zero_arr_row = np.zeros((1, a_obtained.shape[1]))
# [[0. 0. 0. 0.]]

a_obtained = np.insert(a_obtained, desired_cols_ids, zero_arr_row, axis=0)
# [[ 1  2  3  4]
#  [ 5  6  7  8]
#  [ 0  0  0  0]
#  [ 9 10 11 12]
#  [13 14 15 16]
#  [ 0  0  0  0]]

zero_arr_col = np.zeros((a_obtained.shape[0], 1))
# [[0.]
#  [0.]
#  [0.]
#  [0.]
#  [0.]
#  [0.]]

a_obtained = np.insert(a_obtained, desired_cols_ids, zero_arr_col, axis=1)
# [[ 1  2  0  3  4  0]
#  [ 5  6  0  7  8  0]
#  [ 0  0  0  0  0  0]
#  [ 9 10  0 11 12  0]
#  [13 14  0 15 16  0]
#  [ 0  0  0  0  0  0]]

为了稍作解释，首先，我们将行插入到数组中(因此它被修改)，然后插入到修改后的数组的列中.