迭代器与生成器
迭代器
迭代器是访问集合的一种方式,是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。
迭代器有两个基本方法 iter() 和 next() 。
字符串、列表或元组对象都可以用于创建迭代器:
list = [1,2,3,4]
tmp = iter(list) #创建迭代器对象
print(next(it)) #输出迭代器的下一个元素
print(next(it))
输出:
1
2
利用for语句对迭代器元素进行遍历:
list = [1,2,3,4]
it = iter(list)
for x in it:
print(x, end=" ")
输出:
1 2 3 4
使用next()函数迭代:
list = [1,2,3,4]
it = iter(list)
while True:
try:
print(next(it),end=" ")
except StopIteration: #StopIteration 异常用于标识迭代的完成,防止出现无限循环的情况
break
输出:
1 2 3 4
生成器
在Python中,使用了yield的函数被称为生成器(generator)。跟普通函数不同的是,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单点理解生成器就是一个迭代器。在调用生成器运行的过程中,每次遇到 yield 时函数会暂停并保存当前所有的运行信息,返回 yield 的值, 并在下一次执行 next() 方法时从当前位置继续运行。调用一个生成器函数,返回的是一个迭代器对象。
以下实例使用 yield 实现1-20的输出:
def work(n):
i=1;
while True:
if(i>n):
return
yield i
i+=1
f = work(20)
while True:
try:
print(next(f),end=" ")
except StopIteration:
break
输出:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
函数
函数是组织好的,可重复使用的,用来实现单一,或相关联功能的代码段。
函数能提高应用的模块性,和代码的重复利用率。你已经知道Python提供了许多内建函数,比如print()。但你也可以自己创建函数,这被叫做用户自定义函数。
定义一个函数
你可以定义一个由自己想要功能的函数,以下是简单的规则:
- 函数代码块以 def 关键词开头,后接函数标识符名称和圆括号 ()。示例:
def 函数名(参数列表): 函数体
- 任何传入参数和自变量必须放在圆括号中间,圆括号之间可以用于定义参数。
- 函数的第一行语句可以选择性地使用文档字符串—用于存放函数说明。
- 函数内容以冒号起始,并且缩进。
- return [表达式] 结束函数,选择性地返回一个值给调用方。不带表达式的return相当于返回 None。
实例:
eg1.
def hello():
print("Hello World!")
eg2.
def area(width, height):
return width*height
w,h=5,6
print("width = ",w," height = ",h," area = ", area(w,h))
参数传递
在Python中,类型属于对象,变量是没有类型的:
a = [1,2,3,4]
a = "hello"
以上代码中,[1,2,3,4]是列表类型,"hello"是字符串类型,而变量a是没有类型的,它仅仅是一个对象的引用(一个指针),可以是指向列表类型对象,也可以是指向字符串类型的对象。
python传不可变对象实例:
def changeint(a):
a=10
print(a) #结果是10
b=2
changeint(b)
print(b) #结果是2
python传可变对象实例:
def changelist(tmp):
tmp.append([1,2,3,4])
print("函数内取值",tmp)
return
mylist = [1,2,3]
changelist(mylist)
print("函数外取值",mylist)
输出:
函数内取值 [1, 2, 3, [1, 2, 3, 4]]
函数外取值 [1, 2, 3, [1, 2, 3, 4]]
参数
调用函数时可使用的正式参数类型:必须参数,关键字参数,默认参数,不定长参数
必须参数
必须参数须以正确的顺序传入函数。调用时的数量必须和声明时的一样。
实例:
def prin(str)
print(str)
return
上面这个函数调用时必须传入一个参数,否则会报错。
关键字参数
关键字参数和函数调用关系紧密,函数调用使用关键字参数来确定传入的参数值。
使用关键字参数允许函数调用时参数的顺序与声明时不一致,因为 Python 解释器能够用参数名匹配参数值。
实例:
def printinfo( name, age ):
"打印任何传入的字符串"
print ("名字: ", name)
print ("年龄: ", age)
return
#调用printinfo函数,参数顺序可以不指定
printinfo( age=19, name="steve95" )
输出:
名字: steve95
年龄: 19
默认参数
调用函数时,如果没有传递参数,则会使用默认参数。以下实例中如果没有传入 age 参数,则使用默认值:
def printinfo( name, age = 20 ):
print ("名字: ", name)
print ("年龄: ", age)
return
#调用printinfo函数
printinfo( age=19, name="steve95" )
print ("------------------------")
printinfo( name="steve95" )
输出:
名字: steve95
年龄: 19
------------------------
名字: steve95
年龄: 20
不定长参数
你可能需要一个函数能处理比当初声明时更多的参数。这些参数叫做不定长参数,和上述 2 种参数不同,声明时不会命名。简单来说就是命名函数时变量前带有 *
或者**
。
加了星号*
的参数会以元组的形式导入,存放所有未命名的变量参数。
实例:
def printinfo( arg1, *vartuple ):
"打印任何传入的参数"
print ("输出: ")
print (arg1)
print (vartuple)
printinfo( 70, 60, 50 )
输出:
70
(60, 50)
如果在函数调用时没有指定参数,它就是一个空元组。我们也可以不向函数传递未命名的变量。
如下实例:
def printinfo( arg1, *vartuple ):
"打印任何传入的参数"
print ("输出: ")
print (arg1)
for var in vartuple:
print (var)
return
printinfo( 10 )
printinfo( 70, 60, 50 )
输出:
10
70
60
50
加了星号**
的参数会以元组的形式导入.
实例:
def printinfo( arg1, **vardict ):
"打印任何传入的参数"
print ("输出: ")
print (arg1)
print (vardict)
printinfo(1, a=2,b=3)
输出:
1
{'a': 2, 'b': 3}
声明函数时,参数中星号 *
可以单独出现,如果单独出现星号 *
后的参数必须用关键字传入。
实例:
def f(a,b,*,c):
return a+b+c
f(1,2,c=3) #输出6
匿名函数
python 使用 lambda 来创建匿名函数。
所谓匿名,意即不再使用 def 语句这样标准的形式定义一个函数。
- lambda 只是一个表达式,函数体比 def 简单很多。
- lambda的主体是一个表达式,而不是一个代码块。仅仅能在lambda表达式中封装有限的逻辑进去。
- lambda 函数拥有自己的命名空间,且不能访问自己参数列表之外或全局命名空间里的参数。
- 虽然lambda函数看起来只能写一行,却不等同于C或C++的内联函数,后者的目的是调用小函数时不占用栈内存从而增加运行效率。
实例:
sum = lambda arg1, arg2: arg1 + arg2
print(sum(10,20)) # 输出30
强制位置参数
Python3.8 新增了一个函数形参语法 /
用来指明函数形参必须使用指定位置参数,不能使用关键字参数的形式。
在以下的例子中,形参 a 和 b 必须使用指定位置参数,c 和 d 可以是位置形参或关键字形参,而 e 和f 要求为关键字形参:
def f(a, b, /, c, d, *, e, f):
print(a, b, c, d, e, f)
正确使用方法:
f(10, 20, 30, d=40, e=50, f=60)
f(10, 20, 30, 40, e=50, f=60)
f(10, 20, c=30, d=40, e=50, f=60)