1、方法
在做算法题时看到用来生成多维数组的方法
nums=[[0]*n for i in range(n)]#初始化一个n*n的零阵
这里[]中由一个表达式和一个for循环组成,for循环代表了nums中依次填充的元素个数,每次填充的元素为表达式的返回值,所以这里如果n=3的话可以得到以下数组
[[0,0,0],[0,0,0],[0,0,0]]
如果把[0]*n,更改为[0]*i则会生成数量依次递增的列表
[[0],[0,0],[0,0,0]]
2、列表推导式
在官网查找了相关文档,是使用列表推导式(列表生成式)
result = [obj.method() for obj in mylist]
除了生成多维列表还有更多其他方面的应用
2.1 生成平方数列
使用列表推导式生成1-9的平方的数列
nums = [i*i for i in range(1,10)]
输出:
[1, 4, 9, 16, 25, 36, 49, 64, 81]
2.2筛选数列
在for循环后可以添加if判断,如果为True则会添加,否则不
生成数列1-9中的偶数数列
nums = [i for i in range(1,10) if i % 2 == 0]
print(nums)
输出:
[2, 4, 6, 8]
删除1-9中的偶数数列
nums = [1,2,3,4,5,6,7,8,9]
nums[::] = [i for i in nums if i % 2 == 0]
print(nums)
输出:
[2, 4, 6, 8]
使用列表推导式进行筛选相比使用显示的迭代更加快速