3、filter
filter()
也接收一个函数和一个序列。和map()
不同的是,filter()
把传入的函数依次作用于每个元素,然后根据返回值是True
还是False
决定保留还是丢弃该元素。相当于一个过滤器作用。例如,在一个list中,删掉偶数,只保留奇数,可以这么写:
def is_odd(n):
return n % 2 == 1 list(filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15]))
# 结果: [1, 5, 9, 15]
这个filter的关键在于实现一个正确的筛选函数。
练习:回数是指从左向右读和从右向左读都是一样的数,例如12321
,909
。请利用filter()
筛选出回数。
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Date : 2018-05-24 12:51:30
# @Author : Chen Jing (cjvaely@foxmail.com)
# @Link : https://github.com/Cjvaely
# @Version : $Id$ # 练习: 回数是指从左向右读和从右向左读都是一样的数,例如12321,909。
# 请利用filter()筛选出回数: def exchange(num):
count = len(str(num)) - 1 # 数字位数3214 -> 4123
sum = 0
m = 0
while num != 0:
sum = sum + (num // (10 ** count)) * \
(10 ** m) # sum = 3 23 123 4123
num = num % (10 ** count) # num = 214 14 4
count = count - 1 # 2 1 0
m = m + 1 # 1 2 3
return sum def is_palindrome(n):
count = len(str(n)) # 数字位数
if count == 1:
return True
elif count % 2 == 0: # 位数为偶数
half1 = n // 10 ** (count // 2) # 数字前半部分
half2 = n % 10 ** (count // 2) # 数字后半部分
if half1 == exchange(half2):
return True
else: # 位数非1 且为奇数
half1 = n // 10 ** ((count + 1) // 2) # 数字前半部分
half2 = n % 10 ** ((count - 1) // 2) # 数字后半部分
if half1 == exchange(half2):
return True # 借鉴切片: return str(n) == str(n)[::-1]
# 测试:
output = filter(is_palindrome, range(1, 1000))
print('1~1000:', list(output))
if list(filter(is_palindrome, range(1, 200))) == [1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 22, 33, 44, 55, 66, 77, 88, 99, 101, 111, 121, 131, 141, 151, 161, 171, 181, 191]:
print('测试成功!')
else:
print('测试失败!')