Python列表推导式的用法
这里记录Python中的列表(List)的一些快捷操作
一、List是什么?
序列是Python中最基本的,也是最常见的数据结构。序列中的每个元素都分配一个数字 - 它的位置,或索引,第一个索引是0,第二个索引是1,依此类推。
二、使用
1.生成List
代码如下(示例):
a = ['Hello', 'World', 1997, 2007]
b = [[1, 2], [3, 4]]
和字符串一样,可以对list进行索引,截取,组合等。
例如:
print(a) # 输出为:['Hello', 'World', 1997, 2007]
print(a[0]) # 输出为:Hello
print(a[0:2]) # 输出为:['Hello', 'world']
print(b[0]) # 输出为:[1, 2]
print(b[0][1]) # 输出为:2
2.list的一些简便代码
2.1 删除List中的多个相同元素
在统计数据时,生成的列表可能包含一些空值,或者一些超过范围的值,这时我们想要批量删除一些相同元素,除了循环利用python自带的remove函数外,更快捷的操作举例如下:
a = [1,2,3,4,0,8,0,0,0,0] # 删除所有0
b = list(filter(lambda x: x != 0, a))
print(b)
>>> [1, 2, 3, 4, 8]
2.2 生成多个空列表
在统计一些参数时候,需要生成多个空List,此时直接办法是:
a = []
b = []
c = []
d = []
...
简洁办法为:
a, b, c, d = [[] for i in range(4)]
该方法不仅可以应用于创建空列表,还可以对列表中的每个元素进行批量操作,例如对列表中的数据取绝对值可以写成:
a = [1, -1, 2, -2]
a_ads = [ads(i) for i in a]
print(a_abs)
>>> [1, 1, 2 ,2]
其中,对列表中的元素去绝对值的操作可以换成任意其他对于元素的操作,这里可以举一反三进行使用
2.3 循环加判断
当我们想计算两列表中某些元素相乘时,最直接的方法如下:
a = [1, 2, 3, 4, 5]
b = [2, 4, 6, 8, 10]
# 将a中>3的数和b中<8的数相乘
result = []
for i in a:
if a>3:
for j in b:
if j<8:
c = i*j
result.append(c)
print(result)
>>> [8, 16, 24, 10, 20, 30]
而如果用列表推导式则一行代码就搞定
result = [i*j for i in a if i>3 for j in b if j<8]
print(result)
>>> [8, 16, 24, 10, 20, 30]
总结
列表推导式的使用增加了代码的可读性,减少代码的行数。对我个人来说,列表推导式比for循环看着更舒服。当然以上仅供参考,如有疑问,欢迎评论!