Python將行轉換爲數組

Question

我有一個包含所有7x7圖像數據的CSV數據集。有7800行和49列。每行代表一個圖像。數據看起來像這樣。 數據集圖像

我需要應用陣列的每個行，這樣我可以重塑所有DATAS爲2D維數組具有形狀像這樣但在7×7。形狀示例：

我正在使用python，所以我應該怎麼做？

Answer 1

我會做這樣的：

pd.Series(df.values.reshape(-1, 7, 7).tolist()).apply(np.array) 

就這麼完成大部分工作的df.values.reshape(-1, 7, 7)。

array([[[101, 62, 20, 9, 125, 160, 144], 
     [214, 66, 0, 25, 170, 93, 51], 
     [209, 11, 56, 80, 204, 156, 189], 
     [ 95, 211, 26, 172, 207, 39, 249], 
     [ 73, 160, 105, 167, 189, 230, 75], 
     [ 49, 116, 109, 168, 215, 186, 35], 
     [ 68, 154, 185, 12, 186, 78, 252]], 

     [[247, 18, 14, 67, 67, 89, 93], 
     [110, 174, 139, 88, 123, 131, 254], 
     [232, 3, 52, 46, 113, 6, 141], 
     [221, 220, 116, 141, 65, 54, 84], 
     [203, 51, 222, 204, 237, 121, 80], 
     [199, 4, 102, 169, 0, 108, 144], 
     [ 23, 249, 127, 20, 226, 2, 214]], 

     [[ 22, 188, 68, 65, 4, 32, 33], 
     [ 63, 241, 120, 147, 253, 162, 255], 
     [158, 228, 18, 56, 193, 232, 183], 
     [153, 0, 188, 131, 42, 106, 156], 
     [ 51, 70, 219, 156, 94, 50, 43], 
     [ 0, 130, 203, 47, 245, 108, 250], 
     [164, 246, 180, 58, 220, 188, 49]]]) 

Answer 2

您可以DF變成矩陣，重塑它：

df.as_matrix().reshape(-1, 7, 7)