Итак, я хочу получить массив с длиной массива индексов, но со значениями из массива значений в порядке массива индексов. Другими словами, я хочу получить что-то вроде этого:
У меня есть массив индексов: [code]test_idxs = np.array([4, 2, 7, 5]) [/code] У меня также есть массив значений (который длиннее): [code]test_vals = np.array([13, 19, 31, 6, 21, 45, 98, 131, 11]) [/code] Итак, я хочу получить массив с длиной массива индексов, но со значениями из массива значений в порядке массива индексов. Другими словами, я хочу получить что-то вроде этого: [code]array([21, 31, 131, 45]) [/code] Я знаю, как это сделать в цикле, но не знаю, как добиться этого с помощью инструментов Numpy.
У меня проблема с балансировкой строк матрицы CSR. Мне нужно разделить матрицу по строкам на несколько графических процессоров, чтобы каждый графический процессор имел почти одинаковое количество ненулевых элементов.
Для ясности: у меня есть...