Как найти значение в многомерном массиве на основе индекса

Floris спросил: 11 марта 2019 в 07:31 в: python

У меня есть два многомерных массива одинакового размера. Для простоты я сделал их со случайными значениями, но они коррелированы. Из X-массива мне нужно найти максимальное значение на строку. Из массива Y мне нужно значение с соответствующим индексом как максимальное значение из массива X.

import numpy as npX_splitted = np.random.random_sample([517,56])
Y_splitted = np.random.random_sample([517,56])rows = len(Y_splitted[0])
colums = len(Y_splitted)X_max = np.zeros(colums)
index = np.zeros(colums)
Y_corr = np.zeros(colums)for i in range(colums):
    X_max[i] = max(X_splitted[i])
    index[i] = (np.asarray(X_splitted[i].argmax()))
    index = index.astype(int)

Мне удалось найти максимумы массива X и их соответствующие показатели. Однако я не могу сопоставить значения массива Y с этими индексами.

0 ответов