Python学习笔记--高级特性

---

本文为廖雪峰Python教程的学习笔记
具体内容，可参考如下链接：
http://www.liaoxuefeng.com/

---

1. 切片(Slice)

从list, tuple或string中截取一段出来。
[i : j : k]表示在列表中i <= index < j的开区间中以步长为k取出部分元素，步长默认为1，如果k < 0，表示逆序切片。

L = list(range(100))
#取出L[0], L[1], L[2]
L[0:3]

2. 迭代

不使用下标，遍历任何可迭代对象。如list, tuple, dict, string.

#迭代dict类型
#默认对key迭代
for key in d:
    print(key)
#迭代value
for value in d.values():
    print(value)
#同时迭代key, value
for key, value in d.items():
    print(key, value)

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

from collections import Iterable
#判断对象是否可迭代，返回True或False
isinstance(obj, Iterable)

如果使用下标循环，函数enumerate()可以把list变为索引-元素对，可以在for循环中迭代索引和元素。

L = 'abcdefg'
for i, value in enumerate(L):
    print(i, value)

3. 列表生成式(List Comprehensions)

用来创建list的生成式。
元素在前，用for语句生成。可以加上if判断，还可以双层循环。

#for语句，生成结果[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
[x * x for x in range(1, 11)]
#if语句，仅保留偶数的平方，生成结果[4, 16, 36, 64, 100]
[x * x for x in range(1, 11) if x % 2 == 0]
#双层循环，生成全排列，生成结果['aX', 'aY', 'aZ', 'bX', 'bY', 'bZ', 'cX', 'cY', 'cZ']
[m + n for m in 'abc' for n in 'XYZ']
#三层循环，全排列，结果['aX1', 'aX2', 'aY1', 'aY2', 'bX1', 'bX2', 'bY1', 'bY2']
[m + n + q for m in 'ab' for n in 'XY' for q in '12']

列出当前目录下的所有文件和目录名。
os.listdir()可以列出文件和目录。

#导入os模块
import os
dirs = [dir for dir in os.listdir('.')]
print(dirs)

把list中所有字符串变为小写。

L = ['A', 'b', 'C', 'D']
print([e.lower() for e in L])
print(L)

输出结果为：

['a', 'b', 'c', 'd']
['A', 'b', 'C', 'D']

列表生成式新生成一个列表，并不改变原列表的值。
如果list中既包含字符串，又包含整数，由于非字符串类型没有lower()方法，所以列表生成式会报错，可以在if语句中用isinstance()函数判断元素是否为字符类型，再转为小写。

L = ['Hello', 'World', 18, 'Apple', None]
[s.lower() for s in L if isinstance(s, str)]
#结果为['hello', 'world', 'apple']

4. 生成器(generator)

生成器不必创建完整list，可以边循环边计算出要使用的list元素，节省了大量的空间。
把列表生成式的[]改为(),就可以创建一个generator。
可以直接打印出list的每个元素，generator的通过next()获得下一个返回值。
generator保存的是算法，每次调用next(g)，就计算出g的下一个元素的值，直到计算到最后一个元素，没有更多的元素时，抛出StopIteration的错误。
generator是可迭代对象，可以使用for循环打印，使用for循环迭代，不用担心没有元素可打印时出现StopIteration错误。。

l = [2 * x for x in range(1, 6)]
g = (2 * x for x in range(1, 6))
print(l)
#直接打印generator会报错
print(g)
print(next(g))
# print(next(g))
print('for')
#for循环接着当前已经打印过的元素打印
for n in g:
    print(n)

结果如下：

[2, 4, 6, 8, 10]
<generator object <genexpr> at 0x01D81C10>
2
for
4
6
8
10

如果一个函数定义中包含yield关键字，那么这个函数就不再是一个普通函数，而是一个generator，yield语句代替print语句。
例如，打印斐波那契数列的前n个数

#打印斐波那契数列的前n个数
def fib(max):
    n, a, b, = 0, 0, 1
    while n < max:
        yield(b)
        a, b = b, a+b
        n = n + 1
    return 'done'

f = fib(5)
#可以使用for语句打印f中的所有元素
# for num in f:
#   print(num)
print(next(f))
print(next(f))
print(next(f))

每次调用next()函数，相当于函数fib()执行到yield语句，并打印结果，调用下一个next时，从上次的yield处继续执行，遇到return语句或执行到最后一行语句时，结束generator。结果为：

1
1
2

但是用for循环调用generator时，拿不到generator的return语句的返回值。如果想要拿到返回值，必须捕获StopIteration错误，返回值包含在StopIteration的value中.
再例如，打印杨辉三角的前n行：

def triangles():
    prel = []
    while True:
        newl = []
        a = 0
        for b in prel:
            newl.append(a+b)
            a = b
        newl.append(1)
        prel = newl
        yield(newl)
#打印杨辉三角的前10行
n = 0
for t in triangles():
    print(t)
    n = n + 1
    if n == 10:
        break

结果如下：

[1]
[1, 1]
[1, 2, 1]
[1, 3, 3, 1]
[1, 4, 6, 4, 1]
[1, 5, 10, 10, 5, 1]
[1, 6, 15, 20, 15, 6, 1]
[1, 7, 21, 35, 35, 21, 7, 1]
[1, 8, 28, 56, 70, 56, 28, 8, 1]
[1, 9, 36, 84, 126, 126, 84, 36, 9, 1]

5. 迭代器

可以直接作用于for循环的对象统称为可迭代对象：Iterable。
可迭代的数据类型包括集合数据类型（如list、tuple、dict、set、str等）和生成器。
凡是可作用于for循环的对象都是Iterable类型；
凡是可作用于next()函数的对象都是Iterator类型，它们表示一个惰性计算的序列，只有在需要返回下一个数据是才会计算。
Iterator可以表示无限大的数据流，但list不可以。
集合数据类型如list、dict、str等是Iterable但不是Iterator，不过可以通过iter()函数获得一个Iterator对象。

from collections import Iterable
from collections import Iterator
L = [1, 2, 3]
G = (x for x in range(1, 4))
#list是可迭代对象但不是迭代器
print(isinstance(L, Iterable))  #True
print(isinstance(L, Iterator))  #False    
#生成器既是可迭代对象又是迭代器
print(isinstance(G, Iterable))  #True
print(isinstance(G, Iterator))  #True
#iter()函数将可迭代对象转换为迭代器
L1 = iter(L)
print(isinstance(L, Iterator))  #False
print(isinstance(L1, Iterator)) #True