numpy按列连接两个维数不同的数组方式
合并两个维数不同的ndarray
假设我们有一个3×2 numpy数组:
x = array(([[1,2], [3, 4], [5,6]]))
现在需要把它与一个一维数组:
y = array(([7, 8,9]))
通过将其添加到行的末尾,连接为一个3×3 numpy数组,如下所示:
array([[1,2,7], [3,4,8], [5,6,9]])
在numpy中按列连接的方法是:
hstack((x,y))
但是这不行,会报错:
ValueError: arrays must have same number of dimensions
解决方法有两种:
方法一:
>>> x = np.array([[1,2],[3,4],[5,6]]) >>> y = np.array([7,8,9]) >>> np.hstack((x, np.array(([y])).T )) array([[1, 2, 7], [3, 4, 8], [5, 6, 9]])
方法二:
>>> x = np.array([[1,2],[3,4],[5,6]]) >>> y = np.array([7,8,9]) >>> np.column_stack((x,y)) array([[1, 2, 7], [3, 4, 8], [5, 6, 9]])
以上这篇numpy按列连接两个维数不同的数组方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。