У меня есть 2D-массив, и мне нужно получить другой 2D-массив с выбранными строками.
Это для механизма рекомендаций, где я выполнил разложение по сингулярным значениям, и теперь мне нужно получить только некоторые конкретныестроки из матрицы U.У меня есть индекс этих строк, но я не знаю, как «изменить» эту матрицу на основе строк индекса.
Чтобы упростить это, я укажу ниже, что я хочу сделать с базовым примером.
IN
a=np.diag(np.arange(14))
rows=np.array([0,3,5])
OUT
#expected output
[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 0, 0]]
Есть ли поддержка?