语法
1.普通
|
1
|
[expression for target in iterable] |
2.带条件
|
1
|
[expression for target in iterable if condition] |
3.嵌套
|
1
2
3
4
|
[expression for target1 in iterable1 if condition1 for target2 in iterable2 if condition2 ... for targetN in iterableN if conditionN] |
实例
|
1
2
3
4
5
|
x = [1, 2, 3, 4, 5]x = [i * 2 for i in x]print(x)# [2, 4, 6, 8, 10] |
|
1
2
3
4
|
y = [i for i in range(10)]print(y)# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] |
|
1
2
3
4
5
6
7
8
9
10
|
# 获取矩阵第二列matrix = [ [1, 2, 3], [3, 5, 6], [7, 8, 9]]col2 = [row[1] for row in matrix]print(col2)# [2, 5, 8] |
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
# 获取矩阵主对角线元素matrix = [ [1, 2, 3], [3, 5, 6], [7, 8, 9]]diag = [matrix[i][i] for i in range(len(matrix))]print(diag)diag2 = [matrix[i][len(matrix) - i - 1] for i in range(len(matrix))]print(diag2)# [1, 5, 9]# [3, 5, 7] |
|
1
2
3
4
5
6
7
8
|
# 创建内嵌列表a = [[0] * 3 for i in range(3)]print(a)a[1][1] = 5print(a)# [[0, 0, 0], [0, 0, 0], [0, 0, 0]]# [[0, 0, 0], [0, 5, 0], [0, 0, 0]] |
|
1
2
3
4
5
|
# 20以内偶数列表b = [i for i in range(20) if i % 2 == 0]print(b)# [0, 2, 4, 6, 8, 10, 12, 14, 16, 18] |
|
1
2
3
4
5
6
|
# 筛选F开头的单词words = ['Great', 'FishC', 'Brilliant', 'Excellent', 'Fantistic']res = [word for word in words if word.startswith('F')]print(res)# ['FishC', 'Fantistic'] |
|
1
2
3
4
5
6
7
8
9
10
|
# 展开二维列表matrix = [ [1, 2, 3], [4, 5, 6], [7, 8, 9]]flatten = [col for row in matrix for col in row ]print(flatten)# [1, 2, 3, 4, 5, 6, 7, 8, 9] |
等价于
|
1
2
3
4
|
flatten = []for row in matrix: for col in row: flatten.append(col) |
|
1
2
3
4
5
|
# 笛卡尔积fulljoin = [x + y for x in '12345' for y in 'abcde']print(fulljoin)# ['1a', '1b', '1c', '1d', '1e', '2a', '2b', '2c', '2d', '2e', '3a', '3b', '3c', '3d', '3e', '4a', '4b', '4c', '4d', '4e', '5a', '5b', '5c', '5d', '5e'] |
|
1
2
3
4
|
ans = [[x, y] for x in range(10) if x % 2 == 0 for y in range(10) if y % 3 == 0]print(ans)# [[0, 0], [0, 3], [0, 6], [0, 9], [2, 0], [2, 3], [2, 6], [2, 9], [4, 0], [4, 3], [4, 6], [4, 9], [6, 0], [6, 3], [6, 6], [6, 9], [8, 0], [8, 3], [8, 6], [8, 9]] |
总结
本篇文章就到这里了，希望能够给你带来帮助，也希望您能够多多关注服务器之家的更多内容!
原文链接：https://blog.csdn.net/cold___play/article/details/123638695
