Существует ли векторизованный способ добавления/индексирования значений из nd_array 2 на основе значений компонента r/c в nd_array 1?

Я хочу векторизовать некоторый код, который у меня есть, но не знаю, как к нему подойти и возможно ли это вообще. Вот что у меня есть до сих пор:

arr_row = [0, 1, 2, ..., n] # Array of size n with random integers
arr_col = [0, 1, 2, ..., n] # Array of size n with random integers
for i in range(n):
     r = arr_row[i]
     c = arr_col[i]
     result = n_by_n_table[r,c]

Итак, коротко о вышеизложенном. У меня есть два массива, которые являются n элементами, я перебираю от i до n, получая элемент в расположении строки/столбца, полученного из моих строк/столбцов. Я получаю этот результат из гарантированной матрицы n x n. Я надеюсь, что смогу векторизовать это для повышения производительности. Вот что я имел в виду:

vectorized = nd_array((n, 3))
vectorized[:, 0] = [0, 1, 2, ..., n] # Likely using np.rand func or something here.
vectorized[:, 1] = [0, 1, 2, ..., n] # Same as above
vectorized[:, 2] = n_by_n_table[vectorized[:,0], vectorized[:,1]]

Возможно ли это и какие-либо проблемы с выполнением вышеизложенного?