python笔记-切片、迭代、列表生成式、生成器、迭代器

2018-06-26 135次浏览 发表评论

切片

取一个listtuple的部分元素,倒数第一个元素的索引是-1

L = list(range(100))

#取前3个元素
print(L[0:3])

#从索引1开始,取出2个元素
print(L[1:3])

#取倒数2个元素
print(L[-2:])

#前10个数,每两个取一个
print(L[:10:2])

#所有数,每5个取一个
print(L[::5])

tuple也是一种list,唯一区别是tuple不可变。因此,tuple也可以用切片操作,只是操作的结果仍是tuple

tup = tuple(range(100))
print(tup[1:10])

字符串也可以看成一种list,也可以进行切片操作:

st = 'abcdefg'
print(st[0:3])
print(st[::2])

使用切片功能写一个去除字符串首位的空格函数:

def trim(s):
    while s[:1] == ' ':
        s = s[1:]
    while s[-1:] == ' ':
        s = s[:-1]
    return(s)

迭代

在Python中,迭代是通过for ... in来完成的:

d = {'a': 1, 'b': 2, 'c': 3, 'd': 4, 'e': 5}

#默认打印dict的key
for k in d:
    print(k)

#打印dict的value
for v in d.values():
    print(v)

#打印key和value
for k, v in d.items():
    print(k, v)

只要作用于一个可迭代对象for循环就可以正常运行。

通过collections模块的Iterable类型判断一个对象是否是可迭代对象:

from collections import Iterable

print(isinstance('abc', Iterable))    #字符串是可迭代对象
True

print(isinstance([1,2,3], Iterable))  #列表是可迭代对象
True

print(isinstance(123, Iterable))      #整数不是可迭代对象
False

Python内置的enumerate函数可以把一个list变成索引-元素对,这样就可以在for循环中同时迭代索引和元素本身:

for i, v in enumerate(['A', 'B', 'C']):
    print(i, v)

for循环里,可同时引用两个变量:

for x, y in [(1, 1), (2, 4), (3, 9)]:
    print(x, y)

列表生成式

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

举例,要生成一个[1x1, 2x2, 3x3, ..., 10x10]列表方法?

for循环迭代:

L = []
for x in range(1, 11):
    L.append(x * x)
print(L)

用列表生成式:

L2 = [x * x for x in range(1, 11)]
print(L2)

#加上if判断,可以筛选出仅偶数的平方
L2 = [x * x for x in range(1, 11) if x % 2 == 0]
print(L2)

#使用两层循环,可以生成全排列
print([m + n for m in 'ABC' for n in 'XYZ'])

列出当前目录下的所有文件和目录名:

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

使用两个变量来生成list

d = {'x': 'A', 'y': 'B', 'z': 'C'}
print([k + '=' + v for k, v in d.items()])

把一个list中所有的字符串变成小写:

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

生成器

在Python中,这种一边循环一边计算的机制,称为生成器:generator。要创建一个generator,有很多种方法。

第一种方法很简单,只要把列表生成式的[]改成(),就创建了一个generator

g = (x * x for x in range(10))

可以通过next()函数获得generator的下一个返回值:

g = (x * x for x in range(10))

#生成器是可迭代对象,所以可以使用for循环:
for i in g:
    print(i)

第二种方法,如果一个函数定义中包含yield关键字,那么这个函数就不再是一个普通函数,而是一个generator

def odd():
    print('step 1')
    yield 1
    print('step 2')
    yield 3
    print('step 3')
    yield(5)

for i in odd():
    print(i)

迭代器

可以被next()函数调用并不断返回下一个值的对象称为迭代器:Iterator。可以使用isinstance()判断一个对象是否是Iterator对象:

from collections import Iterator
print(isinstance((x for x in range(10)), Iterator))
True

print(isinstance([], Iterator))
False

print(isinstance({}, Iterator))
False

print(isinstance('abc', Iterator))
False

生成器都是Iterator对象,但listdictstr虽然是Iterable,却不是Iterator。把listdictstrIterable变成Iterator可以使用iter()函数:

from collections import Iterator
print(isinstance(iter([]), Iterator))

凡是可作用于for循环的对象都是Iterable类型;

凡是可作用于next()函数的对象都是Iterator类型,它们表示一个惰性计算的序列;

集合数据类型如listdictstr等是Iterable但不是Iterator,不过可以通过iter()函数获得一个Iterator对象。

评论

随便说点,聊聊天~