Нарезка во время выполнения

Question

Может кто-нибудь объяснить мне, как нарезать массив numpy.array во время выполнения? Я не знаю ранг (количество измерений) в «время кодирования».

Минимальный пример:

import numpy as np
a = np.arange(16).reshape(4,4) # 2D matrix
targetsize = [2,3] # desired shape

b_correct = dynSlicing(a, targetsize)
b_wrong = np.resize(a, targetsize)

print a
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]]
print b_correct
[[0 1 2]
 [4 5 6]]
print b_wrong
[[0 1 2]
 [3 4 5]]

И мой безобразный dynSlicing ():

def dynSlicing(data, targetsize):
    ndims = len(targetsize)

    if(ndims==1):
        return data[:targetsize[0]],
    elif(ndims==2):
        return data[:targetsize[0], :targetsize[1]]
    elif(ndims==3):
        return data[:targetsize[0], :targetsize[1], :targetsize[2]]
    elif(ndims==4):
        return data[:targetsize[0], :targetsize[1], :targetsize[2], :targetsize[3]]

Resize () не будет выполнять эту работу, так как он переносит массив перед удалением элементов.

Спасибо, Tebas

Answer 1

Передача кортежа объектов-срезов делает работу:

def dynSlicing(data, targetsize):
    return data[tuple(slice(x) for x in targetsize)]

Answer 2

Простое решение:

b = a[tuple(map(slice,targetsize))]

Answer 3

Вы можете напрямую «изменить» его. Это связано с тем, что массивы разрешают только фон.

Вместо этого вы можете скопировать раздел или, что еще лучше, создать вид нужной фигуры http://www.scipy.org/Tentative_NumPy_Tutorial#head-1529ae93dd5d431ffe3a1001a4ab1a394e70a5f2