filter

理论背景：

filter()函数是 Python 内置的另一个有用的高阶函数，filter()函数接收一个函数 f 和一个list，这个函数 f 的作用是对每个元素进行判断，返回 True或 False，filter()根据判断结果自动过滤掉不符合条件的元素，并返回一个迭代器，可以迭代出所有符合条件的元素。

---

问题描述

请利用filter()过滤出1~100中平方根是整数的数，即结果应该是：1, 4, 9, 16, 25, 36, 49, 64, 81, 100。

我最早写的代码是这样子的：

import math
def is_sqrt(x):
	return type(math.sqrt(x)) == int
for i in filter(is_zhengshu,range(1,101)):
    print(i)

运行完发现没有任何的输出。

我就很奇怪，尝试看一下sqrt函数下的值到底是什么类型的。

data = math.sqrt(9)
print(data)

输出为：3.0

原因分析：

也就是说，sqrt()结果的返回值也是浮点数。

解决方案：

import math
def is_zhengshu(x):
    r = int(math.sqrt(x))
    return r * r == x
for i in filter(is_zhengshu,range(1,101)):
    print(i)