获取从 2 x 1D arrays 构建的 2D 数组元素的访问索引

Question

I have the following prior_total built from prior_fish array and np.arange(5) with num_prior_loop=50

# Declare prior array
prior_start = 1
prior_end = 5
prior_fish = np.logspace(prior_start,prior_end,num_prior_loop)
# Prior total
prior_total = np.stack([prior_fish.T, np.arange(5)])

How to get access to prior_total[i,j] , I mean i for the i-th element of prior_fish and j for the j-element of np.aranage(5)?

I tried: prior_total[i,j] and prior_total([[[0,i],[1,j]]] but this doesn't work.

What might be wrong here?

Answer 1

a = np.logspace(1,5,5)
b = np.arange(5)
c = np.stack([a.T, b])

c[i,j] returns a single element, at col i row j

c[0,i], c[1,j] returns two elements, the i th element of a and the j th element of b