代码之家  ›  专栏  ›  技术社区  ›  KansaiRobot

在简单示例中连接失败

  •  0
  • KansaiRobot  · 技术社区  · 2 年前

    我正在尝试的简单示例 this page

    上面写着:

    arr=np.array([4,7,12])
    arr1=np.array([5,9,15])
    np.concatenate((arr,arr1))
    # Must give array([ 4,  7, 12,  5,  9, 15])
    np.concatenate((arr,arr1),axis=1)
    #Must give 
    #[[4,5],[7,9],[12,15]]
    # but it gives *** numpy.AxisError: axis 1 is out of bounds for array of dimension 1
    

    为什么这个例子不起作用?

    1 回复  |  直到 2 年前
        1
  •  2
  •   Alex P    2 年前

    np.vstack 就是你想要的。注意末尾的转置,这会转换 vstack 的2x3结果转换为3x2数组。

    import numpy as np
    
    arr = np.array([4,7,12])
    arr1 = np.array([5,9,15])
    
    a = np.vstack((arr,arr1)).T
    print(a)
    

    输出

    [[ 4  5]
     [ 7  9]
     [12 15]]