Giải thích từ vòng lặp for sang list comprehension trong Python

List comprehension cung cấp cách ngắn gọn tạo list từ list hiện có. Có thể dùng bất kỳ iterable nào (string, tuple...). Cấu trúc: [expression for item in iterable if condition].

Ví dụ từ string

hello_letters = [letter for letter in 'hello']
print(hello_letters)

Kết quả:

['h', 'e', 'l', 'l', 'o']

Tương đương với:

hello_letters = []
for letter in 'hello':
    hello_letters.append(letter)

Với điều kiện (if)

hello_letters = [letter for letter in 'hello' if letter != 'l']
print(hello_letters)

Kết quả: ['h', 'e', 'o']

Thực hiện phép toán trên biến

squares = [i ** 2 for i in range(1, 6)]
print(squares)

Kết quả: [1, 4, 9, 16, 25]

List comprehension rất expressive, dùng rộng rãi trong Pythonic code.