Numpy数组连接

现有数组A=np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]])和B= np.array([1,1,1,1])，请将数组A和数组B进行水平连接获得新数组C，即C的前4列来源于A，最后一列来源于B。

#注意：数组A和B水平连接要求列数相同，因此B要转换成矩阵

 

#********** Begin **********#

 

def return_values():

 

#1.导入numpy包

 

    import numpy as np

 

#2.定义数组A和数组B

 

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

 

    B=np.array([1,1,1,1]).reshape(4,1)

 

#3.水平连接数组A和数组B

 

    C=np.hstack((A,B))  

 

    return(C)

 

#********** End **********#