除了写循环还有什么更好的办法

要验证 b 是否在 a 中，除了写循环还有什么更好的办法？
a = array([False, False, False, True, True, True, False, False, False, False, False, False])
b = array([True, True, True])
i = 0
while True:
if np.all(b == a[i:i+3]) == True:
print('match')
break
i += 1
else:
print('no match')

false

true

match'

37 replies • 2018-05-23 05:56:50 +08:00

ballshapesdsd

May 21, 2018

(''.join('1' if y else '0' for y in b)) in (''.join('1' if x else '0' for x in a))

widewing

May 21, 2018 via Android

[x in a for x in b]

takato

May 21, 2018

这不就是 KMP 吗？

Arnie97

May 21, 2018 via Android

@widewing 你写的这个东西和楼主想实现的功能有任何关系么…

>>> a = [0, 1]
>>> b = [1, 1, 1, 1, 1]
>>> [x in a for x in b]
[True, True, True, True, True]

wdc63

May 21, 2018

>>> from numpy import *
>>> a = array([False, False, False, True, True, True, False, False, False, False, False, False])
>>> b = array([True, True, True])
>>> [ i for i in b if i not in a ]
[]
>>> b = array([True, True, 123])
>>> [ i for i in b if i not in a ]
[123]

空集为真，非空集为假

wdc63

May 21, 2018

本质上还是循环，算法复杂度是没变的。
如果你的需求是求解集合子集，那不允许出现重复元素，直接转换成 set，用 in 判断是否子集
如果你的需求中包含了元素顺序，那就比较困难了。

wdc63

May 21, 2018

@Arnie97 他写反了

pkookp8

May 21, 2018 via Android

只能拿 b 的第一个去 a 里找，找对了找 b 的下一个是不是 a 的下一个，直到 b 的结尾都匹配，则 b 在 a 中
除了循环还能有什么办法

ipwx

May 21, 2018

先引入 rolling_window： https://gist.github.com/seberg/3866040

然后

import numpy as np

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

l = b.shape[0]
needle = np.equal(rolling_window(a, l), b)
is_matched = np.max(np.sum(needle, axis=-1)) == l

ipwx

May 21, 2018

另外题主，b == a[i:i+3] 这种写法已经 deprecated 了，要用 np.equal

roy2220

May 21, 2018

我来偷个懒：
a = bytes([False, False, False, True, True, True, False, False, False, False, False, False])
b = bytes([True, True, True])
match = a.find(b) >= 0
print(match)

lyusantu

May 21, 2018

那就递归吧

Arnie97

May 21, 2018 via Android

@wdc63 显然需求里包含了元素顺序，不然一共就 True 和 False 两种有啥好判断的

songdg

May 21, 2018

@wdc63 这个我知道，只是不想用纯 Python 写循环，又找不到 numpy 有什么函数可用。

wdc63

May 21, 2018

@songdg 你的需求决定了你的算法只能是循环决定，难道有现成的 numpy 函数它运行的时候就不需要循环了吗？

wdc63

May 21, 2018

@Arnie97 他自己写那段代码就没包含。

wdc63

May 21, 2018

@Arnie97 你的意思如果是要打印出每一个 True False 的话那就用[x in b for x in a]

ipwx

May 22, 2018 via iPhone

@songdg 不点评我那个实现吗

ipwx

May 22, 2018 via iPhone

@wdc63 我那个实现就不用循环

songdg

May 22, 2018

@ipwx 谢谢，你那个有点复杂，一时看不懂，不好意思。

rabbbit

May 22, 2018

我想到的是''.join(map(str, b)) in ''.join(map(str, a))
看了下评论,为什么说 1 楼的代码是错的?

noli

May 22, 2018 via iPhone

false 为 0
True 为 1
bool 数组转换为整型数之后，判断以 b 为模余数是否为 0
就可以了

locktionc

May 22, 2018

楼上这这些人，你们怎么一根筋不知道变通？

```
import numpy as np

a = np.array([False, False, False, True, True, True, False, False, False, False, False, False])
b = np.array([True, True, True])
if str(b)[1:-1] in str(a):
print('b 在 a 中')
```

boboliu

May 22, 2018 via Android

@locktionc 哈哈哈，这操作很骚…

Binb

May 22, 2018 via Android

用字符串匹配喽

lastpass

May 22, 2018 via Android

吓得我赶紧去翻了一下编译原理，这就是最简单的有穷自动机呀。

PythonAnswer

May 22, 2018 via iPhone

Map reduce filter

clearCode1124

May 22, 2018

sets.difference()

ipwx

May 22, 2018 via iPhone

@locktionc 如果不是 bool，这招就不管用了

xxxy

May 22, 2018

楼上说了，用 kmp 可以实现

Raymon111111

May 22, 2018

不就是最长公共子序列吗

congeec

May 22, 2018 via iPhone

@ipwx 老哥这些 gist 里的注释和 doc test 都是你手写的么？

@locktionc 大胸弟你这样会被打死的

ipwx

May 22, 2018

@congeec 这个 gist 不是我写的，是我见过的写得最好的 rolling_window。

我昨天写的 np.max 其实还不太妥当。用 np.any 其实就够了：

is_matched = np.any(np.equal(np.sum(needle, axis=-1), l).astype(np.bool))

locktionc

May 22, 2018

@ipwx 也管用。

ipwx

May 22, 2018

@locktionc 浮点数的话，要用 (a-b) < eps，就用不了你这方法了。

@congeec 上面的 astype(np.bool) 似乎不必要。

lolizeppelin

May 22, 2018 via Android

转 set b-a

songdg

May 23, 2018

谢谢各位热心的 v2exer。