添加一個新的數組多維數組在Python

Question

我有一個numpy的數組，如下所示：

np.array([ 
[23, 12, 4, 103, 87, 0.6], 
[32, 18, 3, 120, 70, 0.6], 
[43, 12, 8, 109, 89, 0.4], 
[20, 13, 7, 111, 77, 0.8] 
]) 

我想要改造這個數組，其中最後一欄變成了自己的陣列，這樣它會看起來像這樣：

np.array([ 
[[23, 12, 4, 103, 87], [0.6]], 
[[32, 18, 3, 120, 70], [0.6]], 
[[43, 12, 8, 109, 89], [0.4]], 
[[20, 13, 7, 111, 77], [0.8]] 
]) 

什麼是最好的方式去做這件事？我對Python比較陌生，嘗試過一些循環，但無濟於事。謝謝！

Answer 1

numpy要求其數組中一致的維數;這會給兩種不同的尺寸。您可以使用兩個單獨的變量（即平行排列）：

X = data[:, :-1] 
y = data[:, -1] 

X = np.array([ 
[23, 12, 4, 103, 87], 
[32, 18, 3, 120, 70], 
[43, 12, 8, 109, 89], 
[20, 13, 7, 111, 77], 
]) 


y = np.array([ 
0.6, 0.6, 0.4, 0.8 
]) 

或者你可以存儲對的列表：

my_list = [(row[:-1], [row[-1]]) for row in data] 
my_list = [ 
([23, 12, 4, 103, 87], [0.6]), 
([32, 18, 3, 120, 70], [0.6]), 
([43, 12, 8, 109, 89], [0.4]), 
([20, 13, 7, 111, 77], [0.8]) 
] 

最好的策略取決於你的使用情況。