Как использовать numpy empty_like

Question

У меня есть следующий массив:

rxn_probability = [1. 0. 0.]

И я хочу создать другой массив, num_rxn, который имеет ту же форму и размер, что и rxn_probability, который содержит ряд реакций, поэтому в этот случай num_rxn будет: [1, 2, 3]. Начиная с 1 и увеличивая, пока он не достигнет того же размера и формы rxn_probability, чтобы при изменении размера rxn_probability размер и форма num_rxn автоматически изменялись.

Пока что я пробовал:

num_rxn = np.array(range(len(rxn_probability + 1)))

(также пытался использовать arange аналогичным образом)

Но это выводит:

[0 1 2]

это не то, что я хочу, потому что оно не начинается в 1 или заканчивается в 3.

Я читал о numpy .empty_like, но я не уверен, что это будет лучшим или правильное решение. Любые идеи?

Приветствия

Answer 1

Вы можете использовать np.arange, а затем изменить форму, чтобы соответствовать форме rxn_probability:

num_rxn = np.arange(1, rxn_probability.size + 1).reshape(rxn_probability.shape)