オーバーレイされたインデックスのスライスでnumpy配列に値を代入

Question

配列の一部のセグメントに値を割り当てたいとします。 セグメントのインデックスがタプル（start_idx、end_idx）になっています。 セグメントは、互いにオーバーレイするか、またはサブセグメントである可能性があります。

a = np.zeros(12) 
segments = np.array([(0, 3), (1, 2), (6, 8), (8, 10)]) 
a[segments] = 1 

結果は次のとおりです。

a 
>> array([1, 1, 1, 1, 0, 0, 1, 0, 1, 0, 1, 0]) 

がどのように私は、この出力を得るために、すべてのセグメントをマスクすることができます。

a 
>> array([1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0]) 

Answer 1

一つのオプションがにありますセグメントを個だけループしてください、および変換範囲実際のインデックスへ：

a = np.zeros(10) 
segments = np.array([(0, 3), (1, 2), (6, 8), (8, 10)]) 

a[[i for s in segments for i in range(*s)]] = 1  
a 
# array([ 1., 1., 1., 0., 0., 0., 1., 1., 1., 1.]) 

Answer 2

を試してみて、この：

a = np.zeros(10) 
segments = np.array([(0, 3), (1, 2), (6, 8), (8, 10)]) 
a[range(3)+range(1,2)+range(6,8)+range(8,10)] = 1 
print (a) 

Answer 3

ここthis postから借用されているという考えを持つ1つのベクトル化されたアプローチだ -

def segment_arr(segments, L): # L being length of output array 
    s = np.zeros(L,dtype=int) 
    stop = segments[:,1]+1 
    np.add.at(s,segments[:,0],1) 
    np.add.at(s,stop[stop<len(s)],-1) 
    return (s.cumsum()>0).astype(int) 

サンプル実行 -

In [298]: segments = np.array([(0, 3), (1, 2), (6, 8), (8, 10)]) 

In [299]: segment_arr(segments, L=12) 
Out[299]: array([1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0]) 

Answer 4

ちょうど言及するが、些細な解決策：for -loopをで使う10とスライスに割り当てます：

import numpy as np 
a = np.zeros(12) 
segments = np.array([(0, 3), (1, 2), (6, 8), (8, 10)]) 

for seg in segments.tolist(): # the "tolist" is just an optimization here, you *could* omit it. 
    a[seg[0]: seg[1]+1] = 1 # or just "seq[1]" if you want to exclude the end point 
print(a) 
# array([ 1., 1., 1., 1., 0., 0., 1., 1., 1., 1., 1., 0.])