numpy配列のリストとリシェイプを組み合わせる

Question

私は次のことを誰かが助けてくれることを願っています。 私は2つの配列のリストを持っています。これらの配列はそれぞれ互いにリンクされていなければなりません。各リストは特定のオブジェクトを表します。 arr1およびarr2は、そのオブジェクトの属性です。例えば ：アレイはarr1で1のように互いに連結されている

import numpy as np 

arr1 = [np.array([1, 2, 3]), np.array([1, 2]), np.array([2, 3])] 
arr2 = [np.array([20, 50, 30]), np.array([50, 50]), np.array([75, 25])] 

、最初の配列は20arr2における最初の配列に属します。私がこの例で探している結果は、サイズが3,4のnumpy配列です。 'columns'は0,1,2,3（arr1の数字に0を加えたもの）を表し、行はarr2の対応する値で埋められます。該当する値がない場合、このセルは0 例にすべきである：

array([[ 0, 20, 50, 30], 
     [ 0, 50, 50, 0], 
     [ 0, 0, 75, 25]]) 

どのように配列のこれら2つのリストを連結し、上記の例に示すように所望のフォーマットでそれらを再構築でしょうか？

多くの感謝！

Answer 1

はここでほとんど*ベクトル化されたアプローチだ - *

lens = np.array([len(i) for i in arr1]) 

N = len(arr1) 
row_idx = np.repeat(np.arange(N),lens) 
col_idx = np.concatenate(arr1) 

M = col_idx.max()+1 
out = np.zeros((N,M),dtype=int) 
out[row_idx,col_idx] = np.concatenate(arr2) 

：ほぼので、スタート時のループの理解のが、それはそこに任意の計算を必要としないとして、それは、計算上無視できるはずです。

Answer 2

ここではforループを使用したソリューションです。各ステップを詳細に示します。

import numpy as np 

arr1 = [np.array([1, 2, 3]), np.array([1, 2]), np.array([2, 3])] 
arr2 = [np.array([20, 50, 30]), np.array([50, 50]), np.array([75, 25])] 

maxi = [] 
for i in range(len(arr1)): 
    maxi.append(np.max(arr1[i])) 
maxi = np.max(maxi) 
output = np.zeros((len(arr2),maxi)) 
for i in range(len(arr1)): 
    for k in range(len(arr1[i])): 
     output[i][k]=arr2[i][k] 

Answer 3

これは、反復の唯一のレベルで、簡単な方法である：私はそれは、この例の@Divakar'sアプローチよりも高速である疑いがある

In [261]: res=np.zeros((3,4),int) 
In [262]: for i,(idx,vals) in enumerate(zip(arr1, arr2)): 
    ...:  res[i,idx]=vals 
    ...:  
In [263]: res 
Out[263]: 
array([[ 0, 20, 50, 30], 
     [ 0, 50, 50, 0], 
     [ 0, 0, 75, 25]]) 

。また、列の数が行の数よりかなり大きい場合は、競争力を維持する必要があります。