numpy blit（將數組的一部分複製到另一個不同大小的數組）

Question

我想將一個數組複製到另一個具有不同大小的數組。 我想是這樣的函數：

blit(destimg,src,dstlocation) 

例如blit(zeros((7,7)),ones((3,3)),(4,4))

將導致

array([[ 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 1., 1., 1.], 
     [ 0., 0., 0., 0., 1., 1., 1.], 
     [ 0., 0., 0., 0., 1., 1., 1.]]) 

陣列src的左上中央現在處於的位置處的陣列destimg的(4,4) 。

如果我沒有blit(zeros((7,7)),ones((3,3)),(5,5))我會得到：

array([[ 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 1., 1.], 
     [ 0., 0., 0., 0., 0., 1., 1.]]) 

數組src不適合在destimg，但其左上角仍然在正確的位置。

Answer 1

你可以只計算相應的切片：

import numpy as np 

def blit(dest, src, loc): 
    pos = [i if i >= 0 else None for i in loc] 
    neg = [-i if i < 0 else None for i in loc] 
    target = dest[[slice(i,None) for i in pos]] 
    src = src[[slice(i, j) for i,j in zip(neg, target.shape)]] 
    target[[slice(None, i) for i in src.shape]] = src 
    return dest 

print(blit(np.zeros((7,7)), np.ones((3,3)), (5, 5))) 

產量

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

和

print(blit(np.zeros((7,7)), np.ones((3,3)), (-1, -1))) 

產量

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

Answer 2

我能找到一個解決方案（有點詳細），必須有一個更優雅的方式，但在這段時間內會做。

from numpy import * 

def blit(dest, src, loc): 
    th,tw=dest.shape 
    sh,sw=src.shape 
    sr = 0 if -loc[0]<0 else -loc[0] 
    fr = sh if loc[0]+sh<=th else sh-(loc[0]+sh-th) 
    sc = 0 if -loc[1]<0 else -loc[1] 
    fc = sw if loc[1]+sw<=tw else sw-(loc[1]+sw-th) 
    loc[0] = max(0,loc[0]) 
    loc[1] = max(0,loc[1]) 
    dest[loc[0]:loc[0]+sh-sr,loc[1]:loc[1]+sw-sc] = src[sr:fr,sc:fc] 

dest = zeros((7,7)) 
src = ones((3,3)) 
loc = [5,5] 
blit(dest, src, loc) 
print dest 

產量：

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

和

dest = zeros((7,7)) 
src = ones((3,3)) 
loc = [-1,-1] 
blit(dest, src, loc) 
print dest 

產生

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