numpy 배열을 사용하여 배열의 인덱스를 분할하는 방법은 무엇입니까?

Question

아마도이 문제가 제기되어 다른 곳으로 전달되었지만 찾지 못했습니다. 각 슬라이스는 또한 a 유효하도록

a = np.arange(100).reshape(10,10) 
b = np.zeros(a.shape) 
start = np.array([1,4,7]) # can be arbitrary but valid values 
end = np.array([3,6,9])  # can be arbitrary but valid values 

start 및 end 모두 유효한 값이 우리가 NumPy와 배열을 가정 .

이

b[:, start:end] = a[:, start:end] #error 

이 구문이 작동하지 않지만, 동등의 : 나는 b에서의 해당 지점에 a에서 하위 어레이의 값을 복사 싶어 더 나은가 궁금

b[:, start[0]:end[0]] = a[:, start[0]:end[0]] 
b[:, start[1]:end[1]] = a[:, start[1]:end[1]] 
b[:, start[2]:end[2]] = a[:, start[2]:end[2]] 

이 방법은 start 및 end 배열을 통해 명시적인 for 루프 대신 사용됩니다.

감사합니다.

Answer 1

우리는 start에 대한 비교의 두 세트와 end 배열을 편집 할 장소의 마스크를 만들 수 broadcasting를 사용하고 단순히 벡터화 된 솔루션을 boolean-indexing으로 할당 할 수 있습니다 -

# Range array for the length of columns 
r = np.arange(b.shape[1]) 

# Broadcasting magic to give us the mask of places 
mask = (start[:,None] <= r) & (end[:,None] >= r) 

# Boolean-index to select and assign 
b[:len(mask)][mask] = a[:len(mask)][mask] 

샘플 실행 -

In [222]: a = np.arange(50).reshape(5,10) 
    ...: b = np.zeros(a.shape,dtype=int) 
    ...: start = np.array([1,4,7]) 
    ...: end = np.array([5,6,9]) # different from sample for variety 
    ...: 

# Mask of places to be edited 
In [223]: mask = (start[:,None] <= r) & (end[:,None] >= r) 

In [225]: print mask 
[[False True True True True True False False False False] 
[False False False False True True True False False False] 
[False False False False False False False True True True]] 

In [226]: b[:len(mask)][mask] = a[:len(mask)][mask] 

In [227]: a 
Out[227]: 
array([[ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9], 
     [10, 11, 12, 13, 14, 15, 16, 17, 18, 19], 
     [20, 21, 22, 23, 24, 25, 26, 27, 28, 29], 
     [30, 31, 32, 33, 34, 35, 36, 37, 38, 39], 
     [40, 41, 42, 43, 44, 45, 46, 47, 48, 49]]) 

In [228]: b 
Out[228]: 
array([[ 0, 1, 2, 3, 4, 5, 0, 0, 0, 0], 
     [ 0, 0, 0, 0, 14, 15, 16, 0, 0, 0], 
     [ 0, 0, 0, 0, 0, 0, 0, 27, 28, 29], 
     [ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], 
     [ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]])