Cách dùng Python extend()

Cách dùng Python extend()

Giả sử bạn sẽ tổ chức tiệc, và có danh sách bạn bè mời. Danh sách này có thể lớn dần khi bạn nghĩ đến người muốn mời. 

Đây là lúc phương thức extend() Python phát huy tác dụng. extend() cho phép thêm một phần tử, hoặc nhiều phần tử, vào cuối list. 

Bài viết này dạy bạn cách dùng extend() Python. Ta dùng ví dụ danh sách mời để minh họa.

Python extend() là gì?

Phương thức extend() Python là một trong các phương thức list của Python. Các phương thức list khác là append() và insert(). Tất cả cho phép thao tác list trong Python. Hãy đào sâu xem tại sao bạn muốn dùng .extend() Python.

Cách Python .extend() hoạt động

Phương thức .extend() mở rộng list Python bằng thêm nhiều mục vào cuối list.  Có hai cách định dạng cú pháp. 

Cách một: đặt mục muốn thêm vào list (gọi là "arguments") trong ngoặc vuông bên trong ngoặc đơn. Xem dòng hai ví dụ, tên mới là arguments trong ngoặc vuông trong ngoặc đơn:

invite_list = ['Shazia', 'Matt', 'Elizabeth']
invite_list.extend(['Janie','Bryce','Jaspar'])
print(invite_list)

Mã trả về:

['Shazia', 'Matt', 'Elizabeth', 'Janie', 'Bryce', 'Jaspar']

Cách hai: khai báo biến mới new_names rồi đặt biến này làm đối số trong ngoặc đơn, như dòng ba ví dụ:

invite_list = ['Shazia', 'Matt', 'Elizabeth']
new_names = ['Janie', 'Bryce', 'Jaspar']
invite_list.extend(new_names)
print(invite_list)

Mã trả về cùng kết quả:

['Shazia', 'Matt', 'Elizabeth', 'Janie', 'Bryce', 'Jaspar']

Khi không nên dùng Python extend()

Nếu bạn chỉ muốn thêm một mục vào cuối list Python, bạn có thể dùng append() thay vì extend(). append() chỉ thêm một mục (như mục list đơn lẻ hoặc list khác) tại một thời điểm. 

Ví dụ, mã này:

invite_list = ['Shazia', 'Matt', 'Elizabeth']
new_names = ['Janie', 'Bryce', 'Jaspar']
invite_list.append(new_names)
print(invite_list)

cho kết quả:

['Shazia', 'Matt', 'Elizabeth', ['Janie', 'Bryce', 'Jaspar']]

Chú ý tên thêm vào (Janie, Bryce, Jaspar) bị bao quanh bởi ngoặc vuông riêng. Nghĩa là append() Python xem chúng như một thực thể đơn. 

Nếu bạn muốn chèn mục list ở vị trí cụ thể, thay vì thêm vào cuối, dùng insert() Python. Trong Python, mục list được gán vị trí. Vị trí là số, bắt đầu từ 0. Do đó, mục trong list ['Shazia', 'Matt', 'Elizabeth'] gán 0, 1, 2, 3. 'Shazia' vị trí 0, 'Matt' vị trí 1, v.v. 

Để dùng insert() với ví dụ danh sách mời, ta nói vị trí mong muốn và tên muốn chèn vào vị trí đó, trong ngoặc đơn dòng hai mã:

invite_list = ['Shazia', 'Matt', 'Elizabeth']
invite_list.insert(0, 'Bryce')
print(invite_list)

Mã trả về kết quả:

['Bryce', 'Shazia', 'Matt', 'Elizabeth']

Ta chèn tên 'Bryce' vào vị trí 0 trong list. 

Kết luận

Phương thức extend() Python dùng thêm mục vào cuối list. Để làm vậy, bạn đặt mục list mới trong ngoặc đơn khi gọi phương thức. Thay vào đó, khai báo biến mới chứa mục list mới rồi đặt biến trong ngoặc đơn. extend() khác append() (chỉ thêm một mục mỗi lần) và insert() (đặt mục ở vị trí cụ thể trong list). 

Bây giờ bạn biết cách dùng extend() Python, bạn có thể thực hành dùng nó, cùng các phương thức list Python khác, lần sau bạn code.