如何在 numpy 中扩展数组？

Question

我有一个数组：a = np.array([[1, 2, 3],               [4, 5, 6]) b = np.array([[7, 8],               [9, 10]])我想用预期的结果扩展数组：c = np.array([[1, 2, 3, 7, 8],               [4, 5, 6, 9, 10])

Answer 1

您可以使用.hstack：

import numpy as np
 
a = np.array([[1, 2, 3],
              [4, 5, 6]])
 
b = np.array([[7, 8],
              [9, 10]])
 
c = np.hstack((a, b))print(c)

输出：

[[ 1  2  3  7  8]
 [ 4  5  6  9 10]]