当值小于零时，如何在子数组中查找索引位置？

Question

对 python 非常陌生，因此为代码的不雅致歉

我有一个带有两个子数组的 numpy 数组：

import numpy as np
p = 0.50
gen_seq = np.random.binomial(1, p, 20).reshape(2, 10)
t_form = np.where(gen_seq == 1, 0.5, -100)
initial_w = np.insert(t_form, 0, 100, axis=1)
d_account = np.cumsum(initial_w, axis=1)

我正在尝试为每个子数组查找低于零的第一个值的索引位置。有什么想法吗？

Answer 1

如果您要获取每个子数组中第一个值的索引，该值小于零；就像下面这样简单，其中a 是有问题的数组。

例子：

# Sub-array 0
(a[0,:] < 0).argmax()
# Sub-array 1
(a[1,:] < 0).argmax()

或者，如果您更喜欢简单的循环：

for i in range(a.shape[0]):
    print((a[i,:] < 0).argmax())

输出：

>>> 6
>>> 3

注意：鉴于原始问题中的代码使用np.random，因此本示例中使用了np.random.seed(73)。