如何将 NumPy 数组附加到 NumPy 数组

Question

我正在尝试填充 NumPy 数组的 NumPy 数组。每次我完成循环的迭代时，我都会创建要添加的数组。然后我想将该数组附加到另一个数组的末尾。例如：

 first iteration
  np.append([], [1, 2]) => [[1, 2]]
next iteration
  np.append([[1, 2]], [3, 4]) => [[1, 2], [3, 4]]
next iteration
  np.append([[1, 2], [3, 4]], [5, 6]) => [[1, 2], [3, 4], [5, 6]]
etc.

我试过使用 np.append 但这会返回一个一维数组，即

[1, 2, 3, 4, 5, 6]

原文由 UBears 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

嵌套数组，使其具有多个轴，然后在使用 append 时指定轴。

 import numpy as np
a = np.array([[1, 2]]) # note the braces
b = np.array([[3, 4]])
c = np.array([[5, 6]])

d = np.append(a, b, axis=0)
print(d)
# [[1 2]
#  [3 4]]

e = np.append(d, c, axis=0)
print(e)
# [[1 2]
#  [3 4]
#  [5 6]]

或者，如果您坚持使用列表，请使用 numpy.vstack ：

 import numpy as np
a = [1, 2]
b = [3, 4]
c = [5, 6]

d = np.vstack([a, b])
print(d)
# [[1 2]
#  [3 4]]

e = np.vstack([d, c])
print(e)
# [[1 2]
#  [3 4]
#  [5 6]]

原文由 pml 发布，翻译遵循 CC BY-SA 3.0 许可协议