numpy：排除数组中的单saltuse和双saltuse

Question

我将Python与numpy一起使用。你知道吗

我有numpy数组b：

 b = np.array([True,True,True,False,False,True,True,False,False,False,True,False,True])

我需要分别替换数组[True]和[True,True]上的[False]和[False,False]。 （我需要排除False中数组的单saltuse和双saltuse）

对于此示例：

out= np.array([True,True,True,True, True, True,True,False,False,False,True,True,True])

有人能建议我怎样得到out？

p.S:如果我需要分别替换数组[False]、[False,False]、[False,False,False]和[False,False,False,False]上的[True]、[True,True]、[True,True,True]和[True,True,True,True]？你知道吗

Answer 1

这可能不是解决这个问题的最佳方法，但请看以下内容。。。你知道吗

In [115]: b
Out[115]:
array([ True,  True,  True, False, False,  True,  True, False, False,
       False,  True, False,  True], dtype=bool)

In [116]: l = [(k,len(list(g))) for k, g in itertools.groupby(b)]

In [117]: l
Out[117]:
[(True, 3),
 (False, 2),
 (True, 2),
 (False, 3),
 (True, 1),
 (False, 1),
 (True, 1)]

In [118]: l2 = [(True, x[1]) if x[1] in [1,2] else x for x in l]

In [119]: l2
Out[119]: [(True, 3), (True, 2), (True, 2), (False, 3), (True, 1), (True, 1), (True, 1)]

In [120]: l3 = [[x[0]] * x[1] for x in l2]

In [121]: l3
Out[121]:
[[True, True, True],
 [True, True],
 [True, True],
 [False, False, False],
 [True],
 [True],
 [True]]

In [122]: l4 = [x for x in itertools.chain(*l3)]

In [123]: l4
Out[123]:
[True,
 True,
 True,
 True,
 True,
 True,
 True,
 False,
 False,
 False,
 True,
 True,
 True]

In [124]: out = np.array(l4)

In [125]: out
Out[125]:
array([ True,  True,  True,  True,  True,  True,  True, False, False,
       False,  True,  True,  True], dtype=bool)