如何将不同形状的np数组组合在一起并按行分隔答案

【问题标题】：How to combine np arrays of different shapes together and separate by row如何将不同形状的np数组组合在一起并按行分隔
【发布时间】：2018-10-14 12:41:21
【问题描述】：

我正在尝试将几个 nparray 组合在一起。它们如下所示：

 array1: [1]
         [0]
         [1]

array2: [ 3  4 14]
        [12  2 30]
        [ 3  4 14]

array3: [ 4  4 14]
        [12  2 30]
        [ 3  4 14]

我想要实现的是：

      (array([3,  4,  14]), 1, array([4, 4, 14))
      (array([12, 2,  30]), 0, array([12,2, 30))
      (array([3,  4,  14]), 1, array([3,4,14]))

我现在正在学习 python，我尝试过使用 np.stack、np.concatenate、但都没有奏效。所以我想知道如何实现所需的输出。

【问题讨论】：

这些看起来像普通的元组。 tuple(zip(array2,array1,array3))?

标签： python arrays python-2.7 numpy

【解决方案1】：

numpy 函数如 np.stack 和 np.concatenate 返回 numpy 数组。

但是，您想要的输出不是 numpy 数组。类型不清楚，因为您没有指定是希望 print 值还是将它们存储在 list 或 tuple 中。

以下是您可以从输入中创建的一些有效对象：

设置

import numpy as np

array1 = np.array([[1], [0], [1]])

array2 = np.array([[3, 4, 14],
                   [12, 2, 30],
                   [3, 4, 14]])

array3 = np.array([[4, 4, 14],
                   [12, 2, 30],
                   [3, 4, 14]])

元组的元组

res = tuple(zip(array2, array1.flatten(), array3))

((array([ 3,  4, 14]), 1, array([ 4,  4, 14])),
 (array([12,  2, 30]), 0, array([12,  2, 30])),
 (array([ 3,  4, 14]), 1, array([ 3,  4, 14])))

元组列表

res = list(zip(array2, array1.flatten(), array3))

[(array([ 3,  4, 14]), 1, array([ 4,  4, 14])),
 (array([12,  2, 30]), 0, array([12,  2, 30])),
 (array([ 3,  4, 14]), 1, array([ 3,  4, 14]))]

单个 numpy 数组

res = np.hstack((array2, array1, array3))

array([[ 3,  4, 14,  1,  4,  4, 14],
       [12,  2, 30,  0, 12,  2, 30],
       [ 3,  4, 14,  1,  3,  4, 14]])

【讨论】：