Python 列表、字典、集合、生成器、迭代器 的一些使用技巧

Python 列表、字典、集合、生成器、迭代器 的一些使用技巧

一、列表（List）使用技巧

1.列表推导式

快速生成列表的方式：

# 基础形式
squares = [x**2 for x in range(10)]

# 带条件
even_squares = [x**2 for x in range(10) if x % 2 == 0]

# 嵌套循环
pairs = [(x, y) for x in range(3) for y in range(3) if x != y]

2.列表切片

a = [1, 2, 3, 4, 5]
a[1:4]    # [2, 3, 4]
a[::-1]   # 反转 [5, 4, 3, 2, 1]
a[::2]    # 步长为2 [1, 3, 5]

3.解包赋值

a, b, *rest = [1, 2, 3, 4, 5]
# a=1, b=2, rest=[3,4,5]

4.常用技巧

# 去重（保持顺序）
unique = list(dict.fromkeys([1,2,2,3,1]))

# 多层嵌套列表展开
flat = [x for row in matrix for x in row]

二、字典（Dict）使用技巧

1.字典推导式

squares = {x: x**2 for x in range(5)}

2.合并与更新

Python 3.9+ 支持新语法：

a = {"x": 1, "y": 2}
b = {"y": 3, "z": 4}
merged = a | b       # {'x': 1, 'y': 3, 'z': 4}

3.安全取值与默认值

person = {"name": "Tom"}
age = person.get("age", 18)  # 默认18

4.反转键值

reversed_dict = {v: k for k, v in person.items()}

5.有序字典（Python 3.7+ 默认保持插入顺序）

for k, v in person.items():
    print(k, v)  # 顺序与插入时一致

三、集合（Set）使用技巧

1.基本集合操作

A = {1, 2, 3}
B = {3, 4, 5}
A | B   # 并集 {1,2,3,4,5}
A & B   # 交集 {3}
A - B   # 差集 {1,2}
A ^ B   # 对称差 {1,2,4,5}

2.集合推导式

unique_squares = {x**2 for x in range(10)}

3.判断子集/超集

A <= B  # A 是否为 B 的子集
A >= B  # A 是否为 B 的超集

4.去重 + 保留顺序

lst = [1,2,3,2,1]
# 用 set 无法保留顺序，这个技巧更优雅。
unique = list(dict.fromkeys(lst))

四、生成器（Generator）使用技巧

生成器是一种 惰性迭代对象，只有取值时才计算，节省内存。

1.基础生成器

def count_up_to(n):
    count = 1
    while count <= n:
        yield count
        count += 1

2.生成器表达式

gen = (x**2 for x in range(10))
next(gen)  # 取下一个值

3.与普通列表对比

列表：一次性生成所有值，占内存

生成器：按需生成（节省内存）

4.双向通信（send）

def greeter():
    name = yield "What's your name?"
    yield f"Hello {name}!"

g = greeter()
print(next(g))        # "What's your name?"
print(g.send("Alice")) # "Hello Alice!"

五、迭代器（Iterator）使用技巧

1.可迭代对象与迭代器的关系

lst = [1, 2, 3]
it = iter(lst)     # 转换为迭代器
next(it)           # 1

2.自定义迭代器类

class Counter:
    def __init__(self, low, high):
        self.current = low
        self.high = high
    def __iter__(self):
        return self
    def __next__(self):
        if self.current > self.high:
            raise StopIteration
        self.current += 1
        return self.current - 1

for num in Counter(1, 5):
    print(num)

3.itertools 模块（迭代器工具集）

Python 自带的强大迭代器库：

from itertools import count, cycle, islice

# 无限计数
for i in islice(count(10), 5):
    print(i)  # 10,11,12,13,14

# 无限循环
for i, x in zip(range(5), cycle("AB")):
    print(x)  # A,B,A,B,A

实战组合例子

示例：生成 100 个随机数，去重、排序、生成平方表：

import random

nums = [random.randint(1, 50) for _ in range(100)]
unique_sorted = sorted(set(nums))
result = {x: x**2 for x in unique_sorted if x % 2 == 0}
print(result)

本文章的原文地址
GitHub主页