0
点赞
收藏
分享

微信扫一扫

Python学习:迭代与列表生成式


1.迭代

如果给定一个list或tuple,我们可以通过for循环来遍历这个list或tuple,这种遍历我们称为迭代(Iteration)

在Python中,迭代是通过for…in来完成的。

d = {'a':1,'b':2,'c':3}
for key in d:
print(key)

输出

a
b
c

默认情况下,dict迭代的是key

for value in d.values():
print(value)

输出

1
2
3

如果要迭代value,可以用for value in d.values()

for k,v in d.items():
print(k,v)

输出

a 1
b 2
c 3

如果要同时迭代key和value,可以用for k, v in d.items()

由于字符串也是可迭代对象,因此,也可以作用于for循环:

for ch in 'ABCD':
print(ch)

输出

A
B
C
D

只要是可迭代对象,都可以迭代,不用关心该对象究竟是list还是其他数据类型

可以通过collections模块的Iterable类型判断

from collections import Iterable
print(isinstance('abc', Iterable))#str是否可迭代
print(isinstance([1,2,3],Iterable))#list是否可迭代
print(isinstance(123,Iterable))#整数是否可迭代

输出

True
True
False

如果要对list实现类似Java那样的下标循环怎么办

Python内置的enumerate函数可以把一个list变成索引-元素对

for i,value in enumerate(['a','b','c']):
print(i,value)

输出

0 a
1 b
2

2.列表生成式

列表生成式即List Comprehensions,是Python内置的非常简单却强大的可以用来创建list的生成式。

要生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]可以用list(range(1, 11)):

print(list(range(1, 11)))

输出

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

但如果要生成[1x1, 2x2, 3x3, …, 10x10]怎么做?方法一是循环

'''
学习中遇到问题没人解答?小编创建了一个Python学习交流群:711312441
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
L = []
for x in range(1,11):
L.append(x*x)
print(L)

输出

[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

但是循环太繁琐,而列表生成式则可以用一行语句代替循环生成上面的list:

写列表生成式时,把要生成的元素x * x放到前面,后面跟for循环,就可以把list创建出来

for循环后面还可以加上if判断,比如选出仅偶数的平方:

x = [x*x for x in range(1,11) if x%2==0]
print(x)

输出

[4, 16, 36, 64, 100]

还可以使用两层循环,可以生成全排列:

e = [m + n  for m in 'abc' for n in 'xyz']
print(e)

输出

['ax', 'ay', 'az', 'bx', 'by', 'bz', 'cx', 'cy', 'cz']

运用列表生成式,可以写出非常简洁的代码。例如,列出当前目录下的所有文件和目录名,可以通过一行代码实现:

import os
print([d for d in os.listdir('.')])#os.listdir可以列出文件和目录

for循环可以使用两个或多个变量

c = {'x':1,'y':2,'z':3}
for k,v in c.items():
print(k,v)
#因此列表生成式可以使用两个变量来生成list
x = {'x':'A','y':'B','z':'c'}
print([k + '= '+ v for k ,v in x.items()])

输出

x 1
y 2
z 3
['x= A', 'y= B', 'z= c']

把list所有字符串变成小写

j = ['Hello','Word','IBM','Apple']
print([s.lower() for s in j ])

输出

['hello', 'word', 'ibm', 'apple']


举报

相关推荐

0 条评论