将 numpy 二维数组转换为 3d

Question

我有一个二维数组，来自 csv

这只是数据形状的一个例子

[[ 0  1  2  3  4  5  6  7  8  9]
 [10 11 12 13 14 15 16 17 18 19]
 [20 21 22 23 24 25 26 27 28 29]
 [30 31 32 33 34 35 36 37 38 39]
 [40 41 42 43 44 45 46 47 48 49]
 [50 51 52 53 54 55 56 57 58 59]
 [60 61 62 63 64 65 66 67 68 69]
 [70 71 72 73 74 75 76 77 78 79]
 [80 81 82 83 84 85 86 87 88 89]
 [90 91 92 93 94 95 96 97 98 99]]

我正在学习使用 numpy，我的目标是将该二维数组转换为形状为 (10,2,4) 的 3d 数组

index 0

[[ 0  1  2  3  4  5  6  7  8  9]
 [10 11 12 13 14 15 16 17 18 19]]

index 1

 [[20 21 22 23 24 25 26 27 28 29]
 [30 31 32 33 34 35 36 37 38 39]]

index 2

 [[40 41 42 43 44 45 46 47 48 49]
 [50 51 52 53 54 55 56 57 58 59]]

index 3

 [[60 61 62 63 64 65 66 67 68 69]
 [70 71 72 73 74 75 76 77 78 79]]

index 4

 [[80 81 82 83 84 85 86 87 88 89]
 [90 91 92 93 94 95 96 97 98 99]]

我可以通过使用循环来做到这一点，但我想知道是否有更好的方法

在一列中连接行也可以

我的目标是拟合 keras model，其中单个样本由 dataframe 的多行组成

Answer 1

您可以使用切片和列表理解

>>> src = [[ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9],
 [10, 11, 12, 13 ,14 ,15 ,16 ,17 ,18 ,19],
 [20, 21, 22, 23 ,24 ,25 ,26 ,27 ,28 ,29],
 [30, 31, 32, 33 ,34 ,35 ,36 ,37 ,38 ,39],
 [40, 41, 42, 43 ,44 ,45 ,46 ,47 ,48 ,49],
 [50, 51, 52, 53 ,54 ,55 ,56 ,57 ,58 ,59],
 [60, 61, 62, 63 ,64 ,65 ,66 ,67 ,68 ,69],
 [70, 71, 72, 73 ,74 ,75 ,76 ,77 ,78 ,79],
 [80, 81, 82, 83 ,84 ,85 ,86 ,87 ,88 ,89],
 [90, 91, 92, 93 ,94 ,95 ,96 ,97 ,98 ,99]]

>>> [src[i:i+2] for i in range(0,len(src),2)]
[[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9], [10, 11, 12, 13, 14, 15, 16, 17, 18, 19]],
 [[20, 21, 22, 23, 24, 25, 26, 27, 28, 29],
  [30, 31, 32, 33, 34, 35, 36, 37, 38, 39]],
 [[40, 41, 42, 43, 44, 45, 46, 47, 48, 49],
  [50, 51, 52, 53, 54, 55, 56, 57, 58, 59]],
 [[60, 61, 62, 63, 64, 65, 66, 67, 68, 69],
  [70, 71, 72, 73, 74, 75, 76, 77, 78, 79]],
 [[80, 81, 82, 83, 84, 85, 86, 87, 88, 89],
  [90, 91, 92, 93, 94, 95, 96, 97, 98, 99]]]

希望这能回答你的问题

编辑：我重新阅读了您的问题并寻找@Anshumaan-mishra 提到的评论。 您也可以使用 numpy 重塑。

>>> import numpy as np
>>> src = np.reshape(src,(5, 2, 10))
array([[[ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9],
        [10, 11, 12, 13, 14, 15, 16, 17, 18, 19]],

       [[20, 21, 22, 23, 24, 25, 26, 27, 28, 29],
        [30, 31, 32, 33, 34, 35, 36, 37, 38, 39]],

       [[40, 41, 42, 43, 44, 45, 46, 47, 48, 49],
        [50, 51, 52, 53, 54, 55, 56, 57, 58, 59]],

       [[60, 61, 62, 63, 64, 65, 66, 67, 68, 69],
        [70, 71, 72, 73, 74, 75, 76, 77, 78, 79]],

       [[80, 81, 82, 83, 84, 85, 86, 87, 88, 89],
        [90, 91, 92, 93, 94, 95, 96, 97, 98, 99]]])

我用上面的代码回答，因为我专注于你想要的数据结构