將數組轉換爲另一個數組的結構後的數組陣列

Question

我想將一個數組轉換爲另一個數組數組。我不知道如何做到這一點，這裏是陣列：

orig_array = [[0,1],[4],[3],[],[3,2,6],[]] 
my_array = [2,0,1,3,3,4,5] 

wanted_array = [[2,0],[1],[3],[],[3,4,5],[]] 

我想保留空數組。

感謝

Answer 1

獲取每個元素的長度在orig_array，沿長度值執行cumumlative求和給我們my_array需要被拆分的最後使用np.split實際執行分裂在其中指數。因此，實施將是這個樣子 -

lens = [len(item) for item in orig_array] 
out = np.split(my_array,np.cumsum(lens))[:-1] 

採樣運行 -

In [72]: orig_array = np.array([[0,1],[4],[3],[],[3,2,6],[]]) 
    ...: my_array = np.array([2,0,1,3,3,4,5]) 
    ...: 

In [73]: lens = [len(item) for item in orig_array] 
    ...: out = np.split(my_array,np.cumsum(lens))[:-1] 
    ...: 

In [74]: out 
Out[74]: 
[array([2, 0]), 
array([1]), 
array([3]), 
array([], dtype=int64), 
array([3, 4, 5]), 
array([], dtype=int64)] 

Answer 2

你可以做到以下幾點：

import copy 

def reflect_array(orig_array, order): 
    wanted_array = copy.deepcopy(orig_array) 
    for i, part_list in enumerate(orig_array): 
     for j, _ in enumerate(part_list): 
      wanted_array[i][j] = order.pop() 
    return wanted_array 

試運行：

orig_array = [[0,1],[4],[3],[],[3,2,6],[]] 
my_array = [2,0,1,3,3,4,5] 

print reflect_array(orig_array, my_array) 
# [[2, 0], [1], [3], [], [3, 4, 5], []] 

Answer 3

def do(format, values): 
    if type(format) == list: 
     return [do(v, values) for v in format] 
    else: 
     return values.pop(0) 

print do(orig_array, my_array) 

注意：這會破壞值來自的數組。

Answer 4

In [858]: my_array = [2,0,1,3,3,4,5] 

In [859]: [[my_array.pop(0) for _ in range(len(x))] for x in orig_array] 
Out[859]: [[2, 0], [1], [3], [], [3, 4, 5], []] 

使用b=my_array[:]如果你不想改變my_array。

此操作與@karoly's答案相同;更直接，因爲它只假定一層嵌套。