python使用小技巧

使用heapq从列表中获取最大或最小的元素

import heapq
scores = [51, 33, 64, 87, 91, 75, 15, 49, 33, 82]
print(heapq.nlargest(3, scores))  # [91, 87, 82]
print(heapq.nsmallest(5, scores))  # [15, 33, 33, 49, 51]

把列表中的值作为参数传递给方法

可以使用" * "提取列表中的所有元素：
my_list = [1, 2, 3, 4]
print(my_list)  # [1, 2, 3, 4]
print(*my_list)  # 1 2 3 4
当我们想将列表中的所有元素作为方法参数传递时，这很有用：
def sum_of_elements(*arg):
    total = 0
    for i in arg:
        total += i
    return total
result = sum_of_elements(*[1, 2, 3, 4])
print(result)  # 10

列表推导
你可以使用推导如，让我们将列表中的每个数字都取二次方:

numbers = [1, 2, 3, 4, 5]
squared_numbers = [num * num for num in numbers]
print(squared_numbers)

推导不仅仅局限于使用列表。还可以将它们与字典、集合和生成器一起使用。
让我们看另一个例子，使用字典推导将一个字典的值提升到二阶：

dictionary = {'a': 4, 'b': 5}
squared_dictionary = {key: num * num for (key, num) in dictionary.items()}
print(squared_dictionary)  # {'a': 16, 'b': 25}

把字符串列表转换成一个列表

import ast
def string_to_list(string):
    return ast.literal_eval(string)
string = "[[1, 2, 3],[4, 5, 6]]"
my_list = string_to_list(string)
print(my_list)  # [[1, 2, 3], [4, 5, 6]]