时间戳

首页 > 解决方案 > 如何将不同形状的np数组组合在一起并按行分隔

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

问题描述

我正在尝试将几个 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，但都没有成功。所以我想知道如何实现所需的输出。

标签： pythonarrayspython-2.7numpy

解决方案

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

但是，您想要的输出不是numpy数组。类型不清楚，因为您尚未指定是否希望将print值或存储在 alist或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]])

推荐阅读