'''
1.从数组中按条件移除元素,返回移除的内容,原数组为移除后的结果
示例:
# arr = [1,2,3,4];
# remove(arr,n%2==0)
# 返回[2,4],此时arr为[1,3]
# arr = ["abc","a","d","ad"]
# remove(arr,len(arr[n])>1)
# 返回['abc','ad'],此时arr为['a','d']
'''
# encoding: utf-8
def func(arr,rule):
try:
if rule == "%":
arr_new = []
for i in arr:
if i%2 == 0:
arr_new.append(i)
for i in arr_new:
arr.remove(i)
return arr_new
if rule == "length":
arr_new = []
for i in arr:
if len(i) > 1:
arr_new.append(i)
for i in arr_new:
arr.remove(i)
return arr_new
else:
return "input error"
except Exception as err:
print(err)
return
if __name__=="__main__":
#test case
arr = [1,2,3,4]
arr_rule = "%"
brr = ["abc","ad","a","d"]
brr_rule = "length"
ret = func(arr,arr_rule)
print(ret)
print(arr)
ret = func(brr,brr_rule)
print(ret)
print(brr)
################################
结果输出:
[2, 4]
[1, 3]
['abc', 'ad']
['a', 'd']
################################
小贴士:
正序删除列表中元素时,被删元素后面的值会向前顶,然后导致漏删。
倒序删除元素时,被删元素前面的值不会向后靠,所以可以完整的遍历到列表中所有的元素
【刷题】【Python】基础题-移除元素
最新推荐文章于 2024-10-14 09:08:48 发布