Trích xuất các khóa cụ thể từ từ điển (dictionary) trong Python

Dictionary là cấu trúc dữ liệu phổ biến trong Python. Cách lấy ra một dictionary con chỉ chứa các khóa cho trước.

Sử dụng dictionary comprehension

Lặp qua các khóa của dictionary gốc, chỉ giữ những khóa nằm trong tập hợp cho trước.

Ví dụ

dictA = {'Sun': '2 PM', "Tue": '5 PM', 'Wed': '3 PM', 'Fri': '9 PM'}
print("Given dictionary :", dictA)
res = {key: dictA[key] for key in dictA.keys() & {'Fri', 'Sun'}}
print("Dictionary with given keys is :", res)

Kết quả

Given dictionary : {'Sun': '2 PM', 'Tue': '5 PM', 'Wed': '3 PM', 'Fri': '9 PM'}
Dictionary with given keys is : {'Fri': '9 PM', 'Sun': '2 PM'}

Sử dụng hàm dict()

Truyền một generator expression vào dict() với điều kiện khóa nằm trong danh sách cho trước.

Ví dụ

dictA = {'Sun': '2 PM', "Tue": '5 PM', 'Wed': '3 PM', 'Fri': '9 PM'}
print("Given dictionary :", dictA)
res = dict((k, dictA[k]) for k in ['Fri', 'Wed'] if k in dictA)
print("Dictionary with given keys is :", res)

Kết quả

Given dictionary : {'Sun': '2 PM', 'Tue': '5 PM', 'Wed': '3 PM', 'Fri': '9 PM'}
Dictionary with given keys is : {'Fri': '9 PM', 'Wed': '3 PM'}