列表生成式
# 使用列表生成选择特定的行
my_data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
rows_to_keep = [row for row in my_data if row[2] > 5]
print("Output #1 (list comprehension): {}".format(rows_to_keep))
列表生成式的意义是:对于my_data中的每一行,如果这行中索引位置2的值大于5,则保留这行。
集合生成式
#使用集合生成式在列表中选择出一组唯一的元组
my_data = [(1, 2, 3), (4, 5 ,6), (7, 8, 9), (7, 8, 9)]
set_of_tuples1 = {x for x in my_data}
print("Output #2 (set comprehension): {}".format(set_of_tuples1))
set_of_tuples2 = set(my_data) #内置的set函数更好
print("Output #3 (set function): {}".format(set_of_tuples2))
字典生成式
#使用字典生成式选择特定的键-值对
my_dictionary = {'customer1': 7, 'customer2': 9, 'customer3': 11}
my_results = {key : value for key, value in my_dictionary.items() if value > 10}
print("Output #3 (dictionary comprehension): {}".format(my_results))