如何将 1d numpy 数组附加到 2d numpy 数组 python

Question

我想将数组 [3, 3, 3] 附加到数组 [[1, 1, 1], [2, 2, 2]]，这样它就变成了 [[1, 1, 1], [2, 2, 2], [3, 3, 3]]

这是我的代码：

import numpy as np
arr1 = np.array([[1, 1, 1], 
                 [2, 2, 2]])

arr2 = np.append(arr1, [3, 3, 3])

print (arr2)

而不是打印 [[1, 1, 1], [2, 2, 2], [3, 3, 3]]、

它打印 [1, 1, 1, 2, 2, 2, 3, 3, 3].

我刚接触numpy，不明白为什么二维数组突然变成一维了

Answer 1

import numpy as np
arr1 = np.array([[1, 1, 1], 
                 [2, 2, 2]])

arr2 = np.append(arr1, [[3, 3, 3]], axis=0)

print (arr2)

输出：

[[1 1 1]                                                                                                                                                                                                        
 [2 2 2]                                                                                                                                                                                                        
 [3 3 3]]

Answer 2

使用numpy.vstack:

arr2 = np.vstack((arr1, [3,3,3]))
>>> arr2
[[1 1 1]
 [2 2 2]
 [3 3 3]]

Answer 3

您可以使用 numpy 的标准方法 vstack。

这是我的代码。

初始化二维numpy数组

initial_array = np.array([
    [1, 1, 1], 
    [2, 2, 2]
])

定义要附加到初始数组的数组

new_array = np.array([3, 3, 3])

将新数组附加到初始数组作为行

result = np.vstack((initial_array, new_array))

这是结果

print(result)
array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]])

您可以在 here 阅读更多详细信息。

希望这个回答对您有所帮助。谢谢。

如何将 1d numpy 数组附加到 2d numpy 数组 python

how to append a 1d numpy array to a 2d numpy array python

python

numpy

append