问题描述:
有2个数组如下
a = [3,3,3,4,4,4,5,6,7]
b = [3,3,4,4]
第1题:从数组a中删除所有在数组b中出现过的元素。对于上例来说,a删除结束应该等于 [5, 6, 7].
第2题:实现 c = a - b , c应该等于[3, 4, 5, 6, 7].
先看第1题:
常规的思维大致会这么写代码:
for i in a:
if i in b:
a.remove(i)
这么写是错误的,如果在for循环外面加一句打印语句,则会发现a变成了一个奇怪的数组: [3, 4, 4, 5, 6, 7]
删倒也删了几个,但是又没有删除干净。(原因后面会讲)
那么,第1题的正确的做法是什么呢?
其实正确的做法很简洁。如下:
a = [i for i in a if i not in b]
再看第2题:
在公布第2题的答案之前,首先我们要弄清楚,为什么本文最初的一段代码是错的?原因就是,当做了a.remove(i)之后,遍历数组的游标会继续右移,然而此时因为已经remove过一个元素了,所以,游标应该保持原位置不动,而不是继续右移。知道了这个原理之后,我们就有办法来写 a - b 了:
import copy
c = copy.deepcopy(a)
while len(b) > 0:
if b[0] in c:
c.remove(b[0])
b.remove(b[0])
else:
print("Error: b is not a sub set of a")
break
print(c) # c = a - b
以上这篇Python 实现数组相减示例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持python博客。
Powered By python教程网 鲁ICP备18013710号
python博客 - 小白学python最友好的网站!