摺疊陣列的嵌套陣列

Question

我想要一個陣列的形狀爲(N,)和dtype=object，所有陣列的形狀相同，shape，並創建一個數組與shape == (N,) + shape。我想知道是否有人知道這樣做的最佳方法。這是一個例子。

import numpy as np 
array = np.empty(4, dtype=object) 
array[:] = [np.ones([3, 2])] 
array = np.array(array.tolist()) 
print array.dtype 
# float64 
print array.shape 
# (4, 3, 2) 

Answer 1

如果你已經知道你內心的陣列（這裏，(3,2)）的形狀，可以簡化整個過程

subshape = (3,2) 
a = np.empty(tuple([N,]+list(subshape)), dtype=object) 
a[:] = np.ones(subshape) 

，可以讓你避免不必要的轉換到/從名單。

---

現在，假設你有一個(N,)對象陣列a其中每個元素是一個subshape float數組，你可以做：

a = np.vstack(a) 
a.shape = [N,] + list(subshape) 

或更簡單地說：

a = np.array(a.tolist(), dtype=float) 

的.tolist但轉換可能不是非常有效。