Thêm tuple vào danh sách các bộ dữ liệu bằng Python

Question

Tôi mới dùng python và không biết cách tốt nhất để làm điều này.

Tôi có một danh sách các bộ dữ liệu đại diện cho các điểm và danh sách khác đại diện cho các phần bù. Tôi cần một tập hợp tất cả các kết hợp mà hình thức này tạo thành. Dưới đây là một số mã:

offsets = [(0, 0),(0,-1),(0, 1),(1, 0),(-1, 0)] 
points = [(1, 5),(3, 3),(8, 7)] 

Vì vậy, thiết lập của tôi về điểm kết hợp nên

[(1, 5),(1, 4),(1, 6),(2, 5),(0, 5), 
(3, 3),(3, 2),(3, 4),(4, 3),(2, 3), 
(8, 7),(8, 6),(8, 8),(9, 7),(7, 7)] 

Tôi không thể sử dụng NumPy hoặc bất kỳ thư viện khác.

Answer 1

result = [(x+dx, y+dy) for x,y in points for dx,dy in offsets] 

Để biết thêm, hãy xem list comprehensions.

Answer 2

Khá đơn giản:

>>> rslt = [] 
>>> for x, y in points: 
...  for dx, dy in offsets: 
...   rslt.append((x+dx, y+dy)) 
... 
>>> rslt 
[(1, 5), (1, 4), (1, 6), (2, 5), (0, 5), (3, 3), (3, 2), (3, 4), (4, 3), (2, 3), (8, 7), (8, 6), (8, 8), (9, 7), (7, 7)] 

Chuyển qua các điểm và hiệu số, sau đó xây dựng bộ dữ liệu mới thêm offsets cho điểm.

Answer 3

Nếu bạn không quan tâm đến bản sao trong kết quả:

result = [] 
for ox, oy in offsets: 
    for px, py in points: 
     result.append((px + ox, py + oy)) 

Nếu bạn quan tâm đến bản sao trong kết quả:

result = set() 
for ox, oy in offsets: 
    for px, py in points: 
     result.add((px + ox, py + oy)) 

Answer 4

Cá nhân, tôi thích câu trả lời của Alok. Tuy nhiên, cho người hâm mộ của itertools, tương đương với itertools-based (bằng Python 2.6 và mới hơn) là:

import itertools as it 
ps = [(x+dx, y+dy) for (x, y), (dx, dy) in it.product(points, offsets)] 

Tuy nhiên, trong trường hợp này là giải pháp itertools là không nhanh hơn so với một đơn giản (nó thực sự là một chút chậm hơn bởi vì nó cần phải giải nén mỗi x, y nhiều lần cho mỗi lần bù trừ, trong khi cách tiếp cận đơn giản của Alok là mỗi gói x, y nhưng một lần). Tuy nhiên, itertools.product là một sự thay thế tuyệt vời cho các vòng lặp lồng nhau trong các trường hợp khác, vì vậy, nó đáng để biết về nó! -)