如何获得 numpy 第一个 True Only

Question

我有如下给出的数据数组。

a=np.array([1,2,3,4,5,6,7,8,1,2,3,4,5,6])

我想获得像a>5这样的条件格式

我可以用np.where得到它。 但是 np.where 是连续检查下一个真

np.where(a>5,'T','F')
> array(['F', 'F', 'F', 'F', 'F', 'T', 'T', 'T', 'F', 'F', 'F', 'F', 'F', 'T'], dtype='<U1')

我想跳过连续的真值，比如当a>5到 6 时，它的True值，但对于 7，它也是我不想要的True值。 我只想要第一个。

但是，如果值从 7 下降到 4 并再次上升到 6，那么 6 应该是True

Answer 1

您可以使用np.roll移动值，然后检查原始元素是否为 True 并且移动的元素是否为 False

a_bool = a > 5
a_shifted_bool = np.roll(a_bool, 1)
a_shifted_bool[0] = False # first element is last element of a_bool, which we need to ignore

np.where(a_bool & ~a_shifted_bool, 'T', 'F')
array(['F', 'F', 'F', 'F', 'F', 'T', 'F', 'F', 'F', 'F', 'F', 'F', 'F',
       'T'], dtype='<U1')

Answer 2

我猜@Suraj S 用最佳时间分析股票来买卖股票。

我尝试了其他示例，我认为@ggaurav 回答是一个好方法，我在您的回答中学到了更多。

import numpy as np

a=np.array([1,2,3,4,5,6,7,8,1,2,3,4,5,6,4,5,6,7,8,4,2,3,6,4,7,9,4,1,2,3,2,5])

a_bool = a > 5
a_shifted_bool = np.roll(a_bool, 1)
a_shifted_bool[0] = False # first element is last element of a_bool, which we need to ignore

np.where(a_bool & ~a_shifted_bool, 'T', 'F')

array(['F', 'F', 'F', 'F', 'F', 'T', 'F', 'F', 'F', 'F', 'F', 'F', 'F', 'T', 'F', 'F', 'T', 'F', 'F', 'F', 'F', 'F', 'T', 'F', 'T', 'F', 'F', 'F', 'F', 'F', 'F', 'F'], dtype='<U1')