合并两个维数不同的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按列连接两个维数不同的数组方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

标签:
numpy,连接,维数,数组

免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
评论“numpy按列连接两个维数不同的数组方式”
暂无“numpy按列连接两个维数不同的数组方式”评论...