Скопировать массив в часть другого массива в NumPy

Question

Я хотел бы скопировать меньший массив A в больший массив B, например, так: Очевидный способ сделать это - вычислить, какая часть A поместится в B, и скопировать только этучасть к предварительно рассчитанной части массива назначения B. Это трудоемкая и подверженная ошибкам задача, особенно при больших размерах.Есть ли более простой (то есть без необходимости расчета всех индексов) способ добиться этого?

Уточнение - обычно нужно:

по оси X: вычислить, сколько элементовдля A, подходящего для части B, вычтите это из размера A, пересчитайте начальную и конечную точки (индексы) в B, где они подходят, пересчитайте индексы в A. Повторите во втором измерении (возможно, больше).Вставьте формулу нарезки в виде луонга, которая никоим образом не читается, а затем убедитесь, что ничего не пропустите в процессе.Это то, что я имею в виду под утомительно и подвержено ошибкам .Это можно сделать, но есть ли лучший способ?

Answer 1

Вы можете воспользоваться арифметикой массива NumPy для обработки N измерений так же, как 1 измерение.Кроме того, вычисление срезов для измерения N может быть выполнено с помощью вызовов map(slice, start_indices, end_indices), и после того, как эти срезы будут сгенерированы, фактическое копирование становится однострочным: B[B_slices] = A[A_slices]:

import numpy as np

def copy_from(A, B, A_start, B_start, B_end):
    """
    A_start is the index with respect to A of the upper left corner of the overlap
    B_start is the index with respect to B of the upper left corner of the overlap
    B_end is the index of with respect to B of the lower right corner of the overlap
    """
    A_start, B_start, B_end = map(np.asarray, [A_start, B_start, B_end])
    shape = B_end - B_start
    B_slices = tuple(map(slice, B_start, B_end + 1))
    A_slices = tuple(map(slice, A_start, A_start + shape + 1))
    B[B_slices] = A[A_slices]

A = np.zeros((21,15))
B = np.ones((16,15))
A_start = [11, 5]
B_start =  [6, 0]
B_end =  [15, 9]
copy_from(A, B, A_start, B_start, B_end)
print(B)

приводит к

[[1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
 [1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
 [1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
 [1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
 [1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
 [1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]
 [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1. 1. 1.]]