Chức năng chuyển/giải nén (nghịch đảo của mã zip)?

Question

Tôi có danh sách các mục 2 mục và tôi muốn chuyển đổi chúng thành 2 danh sách, nơi đầu tiên chứa mục đầu tiên trong mỗi bộ và danh sách thứ hai chứa mục thứ hai.

Ví dụ:

original = [('a', 1), ('b', 2), ('c', 3), ('d', 4)] 
# and I want to become... 
result = (['a', 'b', 'c', 'd'], [1, 2, 3, 4]) 

Có một hàm dựng sẵn nào đó?

Answer 1

zip là nghịch đảo của riêng nó! Miễn là bạn sử dụng toán tử * đặc biệt.

>>> zip(*[('a', 1), ('b', 2), ('c', 3), ('d', 4)]) 
[('a', 'b', 'c', 'd'), (1, 2, 3, 4)] 

Cách làm việc này là bằng cách gọi zip với các đối số:

zip(('a', 1), ('b', 2), ('c', 3), ('d', 4)) 

... trừ các đối số được truyền cho zip trực tiếp (sau khi được chuyển đổi sang một tuple), do đó không cần phải lo lắng về số lượng các đối số nhận được quá lớn.

Answer 2

Bạn cũng có thể làm

result = ([ a for a,b in original ], [ b for a,b in original ]) 

Nó nên quy mô tốt hơn. Đặc biệt nếu Python làm tốt trên việc không mở rộng việc hiểu danh sách trừ khi cần thiết.

(Ngẫu nhiên, nó làm cho một 2-tuple (cặp) của danh sách, chứ không phải là một danh sách các hàng, như zip làm.)

Nếu máy phát điện thay vì danh sách thực tế là ok, điều này sẽ làm điều đó:

result = ((a for a,b in original), (b for a,b in original)) 

Máy phát không munch qua danh sách cho đến khi bạn yêu cầu từng phần tử, nhưng mặt khác, chúng giữ tham chiếu đến danh sách gốc.

Answer 3

Nếu bạn có danh sách không có cùng độ dài, bạn có thể không muốn sử dụng zip theo câu trả lời của Patricks. Đây hoạt động:

>>> zip(*[('a', 1), ('b', 2), ('c', 3), ('d', 4)]) 
[('a', 'b', 'c', 'd'), (1, 2, 3, 4)] 

Nhưng với danh sách dài khác nhau, zip truncates từng hạng mục với độ dài của danh sách ngắn:

>>> zip(*[('a', 1), ('b', 2), ('c', 3), ('d', 4), ('e',)]) 
[('a', 'b', 'c', 'd', 'e')] 

Bạn có thể sử dụng bản đồ không có chức năng để điền vào kết quả sản phẩm nào với None:

>>> map(None, *[('a', 1), ('b', 2), ('c', 3), ('d', 4), ('e',)]) 
[('a', 'b', 'c', 'd', 'e'), (1, 2, 3, 4, None)] 

zip() nhanh hơn một chút.

Answer 4

Tôi thích sử dụng zip(*iterable) (đó là đoạn mã bạn đang tìm kiếm) trong các chương trình của tôi như vậy:

def unzip(iterable): 
    return zip(*iterable) 

tôi thấy unzip dễ đọc hơn.

Answer 5

Đó là chỉ một cách khác để làm điều đó nhưng nó đã giúp tôi rất nhiều vì vậy tôi viết nó ở đây:

Có cấu trúc dữ liệu này:

X=[1,2,3,4] 
Y=['a','b','c','d'] 
XY=zip(X,Y) 

Hệ quả là:

In: XY 
Out: [(1, 'a'), (2, 'b'), (3, 'c'), (4, 'd')] 

Các nhiều cách để giải nén nó và quay trở lại bản gốc là cái này theo ý kiến ​​của tôi:

x,y=zip(*XY) 

Nhưng điều này trả về một tuple vì vậy nếu bạn cần một mảng, bạn có thể sử dụng:

xy=(list(x),list(y)) 

Answer 6

>>> original = [('a', 1), ('b', 2), ('c', 3), ('d', 4)] 
>>> tuple([list(tup) for tup in zip(*original)]) 
(['a', 'b', 'c', 'd'], [1, 2, 3, 4]) 

Cung cấp cho một tuple danh sách như trong câu hỏi.

list1, list2 = [list(tup) for tup in zip(*original)] 

Mở hai danh sách.