python高级特性:切片/迭代/列表生成式/生成器-阿里云开发者社区

开发者社区> 杨俊明> 正文

python高级特性:切片/迭代/列表生成式/生成器

简介: 廖雪峰老师的教程上学来的,地址:python高级特性  下面以几个具体示例演示用法: 一、切片 1.1 利用切片实现trim def trim(s): while s[:1] == " " or s[-1:] == " ": # 若第1个元素或最后1个元素为空格 ...
+关注继续查看

廖雪峰老师的教程上学来的,地址:python高级特性 

下面以几个具体示例演示用法:

一、切片

1.1 利用切片实现trim

def trim(s):
    while s[:1] == " " or s[-1:] == " ": # 若第1个元素或最后1个元素为空格
        if s[:1] == " ":
            s = s[1:]
        if s[-1:] == " ":
            s = s[:-1]
    return s

注:字符串可以看做一个list,列表切片的完整写法是 L[start:end],其中end也支持负数,最后一个数用-1表示,第1个数用0表示,如果省略start,表示从0开始,如果省略end,表示到最后1个元素结束。

测试代码:

if trim('hello  ') != 'hello':
    print('测试失败1!')
elif trim('  hello') != 'hello':
    print('测试失败2!')
elif trim('  hello  ') != 'hello':
    print('测试失败3!')
elif trim('  hello  world  ') != 'hello  world':
    print('测试失败4!')
elif trim('') != '':
    print('测试失败5!')
elif trim('    ') != '':
    print('测试失败6!')
else:
    print('测试成功!')

1.2 切片还有第3个参数,即:L[start:end:skip],比如在1-10之间,把奇数、偶数选出来

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

odd = list_1[::2]
even = list_1[1::2]

print(odd)
print(even)

输出:

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

 

二、迭代

list_1 = list(range(1, 11))

print("正向迭代:")
for x in list_1:
    print(x)

print("\n反向迭代:")
for x in reversed(list_1):
    print(x)

print('\n带索引的迭代:')
for m in enumerate(list_1):
    print("list_1[", m[0], "]=", m[1])

dic_1 = {"name": "菩提树下的杨过", "blog": "http://yjmyzz.cnblogs.com/"}

# 字典的迭代
print("\ndict字典迭代1:")
for k in dic_1:
    print("key:", k, ",value:", dic_1[k])

print("\ndict字典迭代2:")
for v in dic_1.values():
    print("value:", v)

print('\ndict字典迭代3:')
for k, v in dic_1.items():
    print("key:", k, ",value:", v)

输出:

正向迭代:
1
2
3
4
5
6
7
8
9
10

反向迭代:
10
9
8
7
6
5
4
3
2
1

带索引的迭代:
list_1[ 0 ]= 1
list_1[ 1 ]= 2
list_1[ 2 ]= 3
list_1[ 3 ]= 4
list_1[ 4 ]= 5
list_1[ 5 ]= 6
list_1[ 6 ]= 7
list_1[ 7 ]= 8
list_1[ 8 ]= 9
list_1[ 9 ]= 10

dict字典迭代1:
key: name ,value: 菩提树下的杨过
key: blog ,value: http://yjmyzz.cnblogs.com/

dict字典迭代2:
value: 菩提树下的杨过
value: http://yjmyzz.cnblogs.com/

dict字典迭代3:
key: name ,value: 菩提树下的杨过
key: blog ,value: http://yjmyzz.cnblogs.com/

  

三、列表生成器

这个老厉害了!比如:要找出1~100内所有奇数的平方数(即:1,3,5... 这些数的平方数)

a = [x ** 2 for x in range(1, 101) if x % 2 == 1 and x ** 2 <= 100]
print(a)

输出:[1, 9, 25, 49, 81]

再比如,打印出当前目录下的所有文件(不考虑递归子目录) 

import os

print([f for f in os.listdir(".")])

小结:写法就是 [... for ... in .. if ...] ,要生成的list项写在for前面,如果迭代时需要指定条件,写在最后的if中。

 

四、生成器(generator)

这是python引入的一个新概念,想想刚才学到的列表生成器:

result1 = [x ** 2 for x in [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]]
print(type(result1), result1)

输出:

<class 'list'> [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

不知道大家想过没有:如果for前面的运算比较复杂(比如:求平方根),而for迭代的列表又很大(比如:1千万个),最终列表生成器肯定运行很慢,会严重影响性能。能不能做到『延时』计算?等到真正要用的时候,再按需计算。这就是生成器(generator)要解决的问题,它与[列表生成器]的区别在于,它只保存计算逻辑(即: 保存算法),并不马上计算结果,真正要用的时候,调用next(g)取出下一个计算结果即可,当然,它也支持迭代。

generator1 = (x ** 2 for x in [1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
print(type(generator1), generator1)

# 取出第1个值
print(next(generator1))
# 取出第2个值
print(next(generator1))

# 打印剩余的值
for x in generator1:
    print(x)

输出:

<class 'generator'> <generator object <genexpr> at 0x1087e7f10>
1
4
9
16
25
36
49
64
81
100

从输出类型上看,它的类型是generator,而非list。单纯从语法上看,只要把"列表生成器"的[],换成()即可。

再来一个复杂点的示例,中学我们都学过"杨辉三角",如果用常规思路,打印出杨辉三角,可以参考下面的代码:

import copy

def triangles(limit):
    first, second = [1], [1, 1]
    print(first)
    if limit > 1:
        print(second)
        if limit == 2:
            return
        x = copy.copy(second)
        while True:
            y = copy.copy(first)
            [y.append(x[i] + x[i + 1]) for i in range(len(x) - 1)]
            y.append(1)
            print(y)
            x = copy.copy(y)
            if len(y) >= limit:
                return


triangles(10)

输出:

[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]

如果,要改写成生成器(generator),该怎么做呢?答案:只要把print(...)的地方,改成yield 即可。

def triangles_generator(limit):
    first, second = [1], [1, 1]
    yield first
    if limit > 1:
        yield second
        if limit == 2:
            return
        x = copy.copy(second)
        while True:
            y = copy.copy(first)
            [y.append(x[i] + x[i + 1]) for i in range(len(x) - 1)]
            y.append(1)
            yield y
            x = copy.copy(y)
            if len(y) >= limit:
                return


g = triangles_generator(10)
# 取出前2个
print(next(g))
print(next(g))
# 剩下的用迭代写法输出
for x in g:
    print(x)

输出与刚才相同,就不重复贴了。关于这个yield,如果还没理解的,可以对比看下面的示例:

def test1():
    return [1, 2, 3]


def test2():
    print("test2=>1")
    yield 1
    print("test2=>2")
    yield 2
    print("test2=>3")
    yield 3


print(test1())
g = test2()
print(next(g))
print(next(g))
print(next(g))

输出:

[1, 2, 3]
test2=>1
1
test2=>2
2
test2=>3
3

test2()遇到yield后,会停下来,保存现场,等待下一次调用next()时,才会继续执行。

作者:菩提树下的杨过
出处:http://yjmyzz.cnblogs.com
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。

版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。

相关文章
【iOS开发】快速生成高斯模糊效果背景
这篇文章是上一篇文章 http://www.jianshu.com/p/c9083a105921 的拓展,不罗嗦了,直接上代码。 //创建毛玻璃效果的背景 func createFrostBackground (img:UIImage,view:UIView) { let w = self.
876 0
如何使用VC6编译sqlite3源码生成动态链接库(版本:sqlite-source-3_6_23_1)
如何使用VC6编译sqlite-source-3_6_23_1 一.版本: 2010-Mar-30  Version3.6.23.1 二.源代码获取: 在官网的http://www.sqlite.
741 0
超越GAN!OpenAI提出可逆生成模型,AI合成超逼真人像
OpenAI最新提出的可逆生成模型Glow,可以使用相对少的数据,快速生成高清的逼真图像,具有GAN和VAE所不具备的精确操作潜在变量、需要内存少等优势。
2406 0
+关注
杨俊明
菩提树下的杨过 http://yjmyzz.cnblogs.com/
1097
文章
0
问答
文章排行榜
最热
最新
相关电子书
更多
文娱运维技术
立即下载
《SaaS模式云原生数据仓库应用场景实践》
立即下载
《看见新力量:二》电子书
立即下载