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

要验证 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 条回复 • 2018-05-23 05:56:50 +08:00

ballshapesdsd

2018-05-21 18:52:37 +08:00

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

widewing

2018-05-21 18:56:15 +08:00 via Android

[x in a for x in b]

takato

2018-05-21 19:08:03 +08:00

这不就是 KMP 吗？

Arnie97

2018-05-21 19:27:53 +08:00 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

2018-05-21 20:33:47 +08:00

>>> 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

2018-05-21 20:37:04 +08:00

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

wdc63

2018-05-21 20:47:27 +08:00

@Arnie97 他写反了

pkookp8

2018-05-21 20:50:19 +08:00 via Android

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

ipwx

2018-05-21 20:51:47 +08:00

先引入 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

2018-05-21 20:55:11 +08:00

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

roy2220

2018-05-21 21:30:03 +08:00

我来偷个懒：
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

2018-05-21 21:43:03 +08:00

那就递归吧

Arnie97

2018-05-21 22:24:01 +08:00 via Android

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

songdg

2018-05-21 22:31:38 +08:00

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

wdc63

2018-05-21 23:11:18 +08:00

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

wdc63

2018-05-21 23:12:03 +08:00

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

wdc63

2018-05-21 23:13:52 +08:00

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

ipwx

2018-05-22 00:34:53 +08:00 via iPhone

@songdg 不点评我那个实现吗

ipwx

2018-05-22 00:35:06 +08:00 via iPhone

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

songdg

2018-05-22 00:58:17 +08:00

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

rabbbit

2018-05-22 01:05:42 +08:00

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

noli

2018-05-22 04:54:38 +08:00 via iPhone

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

locktionc

2018-05-22 07:08:02 +08:00

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

```
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

2018-05-22 07:15:41 +08:00 via Android

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

Binb

2018-05-22 07:56:43 +08:00 via Android

用字符串匹配喽

lastpass

2018-05-22 08:33:38 +08:00 via Android

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

PythonAnswer

2018-05-22 08:36:12 +08:00 via iPhone

Map reduce filter

clearCode1124

2018-05-22 08:46:26 +08:00

sets.difference()

ipwx

2018-05-22 09:52:15 +08:00 via iPhone

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

xxxy

2018-05-22 10:31:34 +08:00

楼上说了，用 kmp 可以实现

Raymon111111

2018-05-22 10:47:06 +08:00

不就是最长公共子序列吗

congeec

2018-05-22 10:50:55 +08:00 via iPhone

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

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

ipwx

2018-05-22 10:58:31 +08:00

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

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

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

locktionc

2018-05-22 11:07:41 +08:00

@ipwx 也管用。

ipwx

2018-05-22 11:12:13 +08:00

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

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

lolizeppelin

2018-05-22 12:16:26 +08:00 via Android

转 set b-a

songdg

2018-05-23 05:56:50 +08:00

谢谢各位热心的 v2exer。