Hướng dẫn list comprehension python

List comprehensions cung cấp một cách ngắn gọn để tạo một danh sách mới. Nó bao gồm dấu ngoặc chứa một biểu thức theo sau nó là một mệnh đề. Các biểu thức có thể là bất cứ điều gì. Kết quả sẽ trả lại một List mới dựa trên đánh giá của mệnh đề trong đó. List comprehensions luôn trả về một danh sách kết quả.

Nếu như chúng ta hay sử dụng như thế này:

new_list = []
for i in old_list:
    if filter[i]:
        new_list.append[expressions[i]]

Thì chúng ta có thể viết lại ngắn gọn như thế này:

new_list = [expression[i] for i in old_list if filter[i]]

Cú pháp

Cú pháp cơ bản tạo 1 list :

[ expression for item in list if conditional ]

Cú pháp ở trên tương đương:

for item in list:
    if conditional:
        expression

chúng ta có thể tạo set hoặc dict:

{ expression for item in list if conditional }

Ví dụ

Để tạo một list gồm các số từ 1 tới 10 chúng ta thường làm:

lst = []
for i in range[1,11]:
	lst.append[i]

print[lst]
[1,2,3,4,5,6,7,8,9,10]

Nếu sử dụng list comprehensions thì công việc rất ngắn gọn:

lst = [i for i in range[1, 11]]

print[lst]
[1,2,3,4,5,6,7,8,9,10]

Sử dụng list comprehensions thay cho map và filter

Nếu như chúng ta chỉ sử dụng một đối số thì list comprehensions sẽ dễ hiểu hơn là chúng ta sử dụng map. Thông thường khi sử dụng map trong python chúng ta thường sử dụng kèm với lambda.

>>> a=  [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 
>>> squares= map[lambda x: x**2, a]
>>> list[squares]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

nếu sử dụng list comprehensions:

>>> a=  [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 
>>> even_squares=[x**2 for x in a]
>>> even_squares
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

rõ ràng khi nhìn vào 2 biểu thức trên thì ta sẽ thấy sử dụng list comprehensions với 1 tham số sẽ dễ hiểu hơn dùng map.

Filter buil-in function có thể được sử dụng kèm với map. FIlter sẽ trả về một danh sách kết quả với các điều kiện là true.

>>> alt	= map[lambda x:	x**2, filter[lambda	x: x%2==0, a]]
>>> list[alt]
[4, 16, 36, 64, 100]

Nhưng khi nhiều vào khối lệnh trên thì nó sẽ rất lằng nhằng và khó đọc. Thay vào đó sử dụng list comprehensions sẽ rất dễ dàng và dễ hiểu:

>>> even_squares =[x**2 for x in a if x%2==0]
>>> even_squares
[4, 16, 36, 64, 100]

Note

List comprehensions cũng hỗ trợ nhiều cấp và điều kiện ở mỗi cấp. Tuy nhiên, chúng ta cũng chỉ nên sử dụng nó ở 2 cấp. Việc sử dụng lồng nhiều cấp độ cũng gây ra tình trạng khó đọc. Chúng ta nên tránh điều đó.

Conclusion

Trên đây là mình đã giới thiệu cách sử dụng List comprehensions trong python. Việc sử dụng nó sẽ giúp code nhìn trông sáng sủa hơn, dễ đọc hơn. Nhưng cũng tránh một số trường hợp. Cái này làm nhiều chũng ta sẽ rút được ra kinh nghiệm. Cảm ơn đã đọc bài.

Tham khảo:

effective python

//www.pythonforbeginners.com/basics/list-comprehensions-in-python

Trong bài này chúng ta sẽ tìm hiểu cách sử dụng list comprehension trong Python, đây là cách tạo ra một list mới dựa vào một list có sẵn dựa vào một điều kiện nào đó.

Bài viết này được đăng tại freetuts.net, không được copy dưới mọi hình thức.

1. List comprehension là gì?

List comprehension là cách tạo ra một list dữ liệu mới dựa vào dữ liệu cũ theo một điều kiện nào đó.

Vậy, công thức chung của list comprehension là:

newlist = [expression for item in iterable if condition == True]

Trong đó:

Bài viết này được đăng tại [free tuts .net]

• expression chính là biểu thức trả về
• iterable là tập dữ liệu iterator
• condition là điều kiện trả về, phần tử nào có condition = True thì biểu thức expression tương ứng sẽ được trả về.

Giả sử mình có một mảng các số như sau:

numbers = [1,3,86,4,54,54,45,65,6,57,67,87,82]

Bây giờ mình muốn tạo một mảng mới từ mảng này, với điều kiện là chỉ lấy các số chẵn mà thôi.

Cách thứ nhất: Sử dụng vòng lặp bình thường.

numbers = [1,3,86,4,54,54,45,65,6,57,67,87,82]

new_numbers = []
for n in numbers:
    if n % 2 == 0:
        new_numbers.append[n]

# kết quả: [86, 4, 54, 54, 6, 82]
print[new_numbers]

Cách thứ hai: Sử dụng list comprehension.

numbers = [1,3,86,4,54,54,45,65,6,57,67,87,82]

new_numbers = [x for x in numbers if x % 2 == 0]

# kết quả: [86, 4, 54, 54, 6, 82]
print[new_numbers]

Bạn có thể không cần phải truyền condition.

Ví dụ: Tạo mảng mới với giá trị gấp đôi mảng cũ.

numbers = [1,3,5,7,9]

new_numbers = [x*2 for x in numbers]

# Kết quả: [2, 6, 10, 14, 18]
print[new_numbers]

2. Ví dụ về list comprehension trong Python

Bây giờ mình sẽ làm thêm một vài ví dụ để bạn dễ hiểu hơn nhé.

Ví dụ: Tạo danh sách các số nguyên tố từ một danh sách có sẵn.

Để giải bài này thì ta cần tạo một hàm kiểm tra số nguyên tố trước, sau đó sẽ sử dụng nó trong phần condition.

# Hàm kiểm tra số nguyên tố
def laSNT[n]:
    if n < 2: return False
    elif n == 2: return True
    elif n % 2 == 0: return False
    else:
        for i in range[3, n, 2]:
            if n % i == 0: return False
    return True

# Dữ liệu
data = [1, 2, 3, 5, 45, 46, 5, 6, 76, 9, 11]

# Lấy danh sách các số nguyên tố trong data
data_snt = [x for x in data if laSNT[x]]
print[data_snt]

Đây chính là cách sử dụng list comprehension để tạo ra danh sách mảng mới trong Python.