Два 2D-массива, нарезка одной строки 2D-массивов и работа со вторым 2D-массивом - PullRequest
Новая
       52

Два 2D-массива, нарезка одной строки 2D-массивов и работа со вторым 2D-массивом

0 голосов
/ 25 октября 2019

Я застрял в точке, где у меня есть два двухмерных массива:

Array1 

test[] = [[0.54131721 0.52305685 0.42921551, 0.37434461 0.52591475 0.36184407]
 [0.53091097 0.3000469  0.39346106, 0.29261769 0.3806552  0.33904193]
 [0.29331853 0.44518117 0.41390863, 0.2510257  0.50481932 0.43607184]]

Array2 

train[] =[[0.5301304,  0.62645837, 0.44524917, 0.40806674 0.46013734 0.61033772]
 [0.43333892 0.46062429 0.56937923, 0.6451305  0.33103777 0.35859095]
 [0.60879428 0.72451976 0.2661216, 0.38850336 0.41685737 0.57226228]]

Я хочу создать функцию, в которойЯ передам полный 2D-массив train [] и одномерный массив test [] (строка за строкой); 

temp_test[] = [0.54131721, 0.52305685, 0.42921551, 0.37434461, 0.52591475, 0.36184407] 
train = [[0.5301304,  0.62645837, 0.44524917, 0.40806674 0.46013734 0.61033772],[0.43333892, 0.46062429, 0.56937923, 0.6451305,  0.33103777, 0.35859095], [0.60879428, 0.72451976, 0.2661216, 0.38850336, 0.41685737 ,0.57226228]]

У меня есть функция, готовая для выполнения одного и того же набора операций на двух 2D-массивах, ноМне нужно сделать это один за другим в строках test []. Вот мой кусок кода для двух 2D массивов: 

def dist_mat(x,y):
    return np.sqrt(np.sum((x- y[:,None])**2, axis=-1))

dist_mat(train,test)
...