在用Python进行coding的世界里,列表是万金油一般的存在。作为Python中最常用的数据结构,熟练掌握列表的各种使用方法,是一名Python初级程序员的日常。而列表解析式(list comprehension, 也成为列表推导式)作为非常pythonic的用法,能够熟练运用则是一个进阶Pythoner的标志。这一次我们就来总结一下列表解析式的各种用法,一次性全部掌握。
首先明确列表解析式的作用,即快速简洁的创建列表。其简洁性在于能用一行代码代替十几行代码,而且不损失任何可读性。而且用列表解析式生成一个列表比普通的for循环加上append方法生成的列表要快很多,这也是为什么提倡使用列表解析式来生成列表的原因。下面我们就按照3个步骤,由浅入深的来解析列表解析式
1.基本格式:
[exp for i in k], 其中:
- exp: 表达式,可以是一个函数,变量为i(也可以与i无关)
- k: k为一个可迭代对象,如列表。
应用举例:
- 一句代码输出一个1到5的立方
- 一句代码创建一个列表,包含10个60-100的随机整数
# 一句代码输出一个1到5的立方
[i**3 for i in range(1,6)]
>>>[1, 8, 27, 64, 125]
# 一句代码创建一个列表,包含10个60-100的随机整数(模拟学生成绩)
import random
[random.randint(60,100) for _ in range(10)]
>>> [76, 89, 62, 83, 61, 80, 89, 99, 76, 78]
上面的例子中,第一个例子的表达式和i相关,第二个和i没有关系。表达式的作用就是生成目标列表中的元素,而后面的for循环决定了生成的列表元素的个数。
2. 加入for循环嵌套
列表推导式中的for循环支持嵌套功能。
举例: 现有3个列表分别保存了顾客的姓名,衣服的颜色,尺码,用一句代码输出所有顾客和衣服颜色尺码的组合
names = ['zhangsan', 'lisi', 'wangba']
color = ['red', 'yellow']
size = ['S', 'M', 'L']
[name + '-' + c + '-' + s for name in names for c in color for s in size]
>>>
['zhangsan-red-S',
'zhangsan-red-M',
'zhangsan-red-L',
'zhangsan-yellow-S',
'zhangsan-yellow-M',
'zhangsan-yellow-L',
'lisi-red-S',
'lisi-red-M',
'lisi-red-L',
'lisi-yellow-S',
'lisi-yellow-M',
'lisi-yellow-L',
'wangba-red-S',
'wangba-red-M',
'wangba-red-L',
'wangba-yellow-S',
'wangba-yellow-M',
'wangba-yellow-L']
上面的代码等价于:
for name in names:
for c in color:
for s in size:
print(name + '-' + c + '-' + 's')
>>>
zhangsan-red-s
zhangsan-red-s
zhangsan-red-s
zhangsan-yellow-s
zhangsan-yellow-s
zhangsan-yellow-s
lisi-red-s
lisi-red-s
lisi-red-s
lisi-yellow-s
lisi-yellow-s
lisi-yellow-s
wangba-red-s
wangba-red-s
wangba-red-s
wangba-yellow-s
wangba-yellow-s
wangba-yellow-s
从这个例子可以充分体会到列表解析式的简洁!
3. 筛选功能
更进一步,列表推导式中for循环后还可以加入if (或者if...else...)进行筛选 。
举例: 一句代码输出0-100内可以被7整除的整数
# 输出1-100内可以被7整除的数:
[i for i in range(1,101) if i%7 == 0]
>>>
[7, 14, 21, 28, 35, 42, 49, 56, 63, 70, 77, 84, 91, 98]
综合上面的案例,我们可以清晰的看到列表推导式的简洁与优雅!同时也体现Python的强大之处。
4. 更多案例
- 现有2个长度一样的数字列表,完成对应位置的元素相乘生成新的列表
a = [i for i in range(1,10)]
b = [i+1 for i in a]
print(a)
print(b)
>>>
[1, 2, 3, 4, 5, 6, 7, 8, 9]
[2, 3, 4, 5, 6, 7, 8, 9, 10]
# 使用列表解析式完成对应位置的元素相乘
[i*j for i, j in zip(a,b)]
>>>
[2, 6, 12, 20, 30, 42, 56, 72, 90]
- 现有一份学生名单,将名单中以S开头且长度大于7的姓名筛选出来
# 先模拟一份名单
name=["Smialexk","Koldboy","Smithweird","Lonosong", \
"Sam","Solway","Podykx","Sa-jack","Sykx","Songarx" ,"Ponkx",'Ponyma']
# 筛选S开头且长度大于7的名字
[i for i in name if len(i) > 7 and i.startswith('S')]
>>>
['Smialexk', 'Smithweird']
其实上面这个例子在引入一些复杂的逻辑就可以完成一个更复杂的事情了,比如现在有一份csv格式的小说,可以用这个逻辑遍历整份csv文件,把想要的人名提取出来在统计词频就可以大概这道这个小说有哪些猪脚了,一般猪脚的名字肯定会反复出现嘛。
OK,本次分享就到这里。更多精彩内容也可以W X搜索,并关注公众号:‘Python数据科学家之路“ ,期待您的到来和我交流