列表生成式/列表推导式:
1、[表达式 for 变量 in 列表]
2、[表达式 for 变量 in 列表 if 条件]
# 生成一个编号1到30号的员工列表
staff_list = [f'staff_{i}' for i in range(1,30)] # [表达式 for 变量 in 列表]
print(staff_list)
# 求列表nums中每个元素的平方,再组成一个新的列表
nums = [1,2,3,4,5]
resultList = [num ** 2 for num in nums] # [表达式 for 变量 in 列表]
print(resultList)
# 求列表nums中每个奇数的平方,再组成一个新的列表
nums = [1,2,3,4,5]
resultList = [num ** 2 for num in nums if num % 2 != 0] # [表达式 for 变量 in 列表 if 条件]
print(resultList)
输出:
['staff_1', 'staff_2', 'staff_3', 'staff_4', 'staff_5', 'staff_6', 'staff_7', 'staff_8', 'staff_9', 'staff_10', 'staff_11', 'staff_12', 'staff_13', 'staff_14', 'staff_15', 'staff_16', 'staff_17', 'staff_18', 'staff_19', 'staff_20', 'staff_21', 'staff_22', 'staff_23', 'staff_24', 'staff_25', 'staff_26', 'staff_27', 'staff_28', 'staff_29']
[1, 4, 9, 16, 25]
[1, 9, 25]