np.column_stack用法

x_vals = np.linspace(0, 10, 5)
#print(x_vals)
[ 0.   2.5  5.   7.5 10. ]
x_vals_column = np.transpose(np.matrix(x_vals))#转化数组为矩阵
ones_column = np.transpose(np.matrix(np.repeat(1, 5))) 
#print(ones_column)
[[1]
 [1]
 [1]
 [1]
 [1]]
 
A = np.column_stack((x_vals_column, ones_column))

#print(A)
[[ 0.   1. ]
 [ 2.5  1. ]
 [ 5.   1. ]
 [ 7.5  1. ]
 [10.   1. ]]
 
b = np.row_stack((x_vals_column, ones_column))
# print(B)
[[ 0. ]
 [ 2.5]
 [ 5. ]
 [ 7.5]
 [10. ]
 [ 1. ]
 [ 1. ]
 [ 1. ]
 [ 1. ]
 [ 1. ]]

猜你喜欢

转载自blog.csdn.net/weixin_38632246/article/details/86713078