明白了-我要通过
some_indexes
作为列表列表,如此麻木地将每个列表广播到
list_of_column_indexes
. 所以这是:
travel_time_vec = np.zeros(...)
# newaxis below tranforms [1, 2, 3] to [[1], [2], [3]]
traveltimes = traveltime_2d_array[np.array(some_indexes)[:, np.newaxis],
list_of_column_indexes]
# get the index of the min time on each row
best_index = np.argmin(traveltimes, axis=1)
travel_time_vec[some_indexes] = traveltimes[:, best_index]
按预期工作,不再循环