添加列数组Python

在Python中，我们可以使用NumPy来添加列数组。NumPy是一个运行速度极快的Python库，用于科学计算和数据处理。

步骤

以下是如何在NumPy中添加列数组的步骤：

1. 导入NumPy库

   import numpy as np
   

2. 创建一个多维数组

   arr = np.array([[1, 2], [3, 4]])
   

3. 创建要添加的新列数组

   new_col = np.array([5, 6])
   

4. 使用hstack()函数将新列数组添加到原始数组中

   new_arr = np.hstack((arr, np.atleast_2d(new_col).T))
   

完整代码：

import numpy as np

arr = np.array([[1, 2], [3, 4]])
new_col = np.array([5, 6])
new_arr = np.hstack((arr, np.atleast_2d(new_col).T))
print(new_arr)

输出结果：

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

解释

我们使用hstack()函数将两个数组水平连接起来（即按列连接）。np.atleast_2d()函数用于将new_col转换为二维数组，并且我们需要使用.T转置它，以便它与原始数组的形状相匹配。最终的新数组为new_arr。

添加列数组的过程非常简单，并且使用NumPy库可以轻松实现。只需要遵循上述步骤即可完成。