Python, cách tốt nhất để viết tổng của hai cho vòng

Question

Thông thường tôi biết chúng ta có thể làm sum([func(x,x) for x in i]), nhưng tôi có một nếu kiểm tra và hai cho vòng, vì vậy cách pythonic nhất để viết mã dưới đây là những gì. bạn có thể giả định rằng sự giống nhau sẽ trả về một số bất kể bạn cung cấp loại nào. Bạn cũng có thể giả định rằng nó sẽ thực sự chỉ nhận được int và ký tự.

x = 0 
if isinstance(a, dict) or isinstance(a, list) or isinstance(a, tuple): 
    for i in a: 
     for j in b: 
      x += similarity (i,j) 

Answer 1

Có lẽ một cái gì đó như thế này:

x=0 
if isinstance(a,(dict,list,tuple)): 
    x=sum(similarity(i,j) for i in a for j in b) 

Hoặc:

x=(sum(similarity(i,j) for i in a for j in b) if isinstance(a,(dict,list,tuple)) 
    else 0) 

Hoặc (giả định rằng một chuỗi, thiết lập hoặc một số loại có thể lặp lại khác không phá vỡ chức năng của bạn vì một số lý do):

try: 
    x=sum(similarity(i,j) for i in a for j in b) 
except TypeError: 
    x=0 

---

Nếu bạn đang tìm cách cụ thể để kiểm tra nếu có điều gì là iterable, bạn có thể làm điều đó theo cách này:

from collections import Iterable 
if isinstance(e, Iterable): 
    ... 

Nếu có một số loại iterable bạn không muốn, phản ứng với những:

if isinstance(e, Iterable) and not isinstance(el, str): 
    # an iterable that is not a string... 

Answer 2

Kể từ khi if không phải là bên trong một vòng lặp, nó không cần phải được bên trong danh sách hiểu khi bạn chuyển đổi nó:

x = 0 
if isinstance(a, dict) or isinstance(a, list) or isinstance(a, tuple): 
    x = sum([similarity(i, j) for i in a for j in b]) 

Trong khi đó, bạn thực sự không cần một danh sách hiểu ở đây, một biểu thức máy phát điện sẽ có tác dụng tương tự mà không cần phải xây dựng một danh sách:

x = 0 
if isinstance(a, dict) or isinstance(a, list) or isinstance(a, tuple): 
    x = sum(similarity(i, j) for i in a for j in b) 

---

Trong khi đó, isinstance có thể mất một tuple các loại để kiểm tra, vì vậy:

x = 0 
if isinstance(a, (dict, list, tuple)): 
    x = sum(similarity(i, j) for i in a for j in b) 

... nhưng thực sự, tôi không nghĩ rằng bạn muốn kiểm tra ở nơi đầu tiên. Miễn là a là bất kỳ lần lặp nào, mã sẽ hoạt động, vì vậy… tại sao tôi nên ép nó là các loại cụ thể? Nếu bạn không muốn một ngoại lệ, chỉ cần xử lý các ngoại lệ:

try: 
    x = sum(similarity(i, j) for i in a for j in b)) 
except TypeError: 
    x = 0 

Answer 3

if type(a) in [dict, list, tuple]: 
    x += sum(similarity(i, j) for i in a for j in b) 

yeah, loại không giống như isistance nhưng nó không phải là một vấn đề đáng kể ..

nếu OP muốn sử dụng đoạn mã đó với bất kỳ thứ gì giống như dict, list hoặc tuples (vì vậy với các lớp con và tất cả) nó thực sự chỉ cần kiểm tra xem a có thể lặp lại hay không.

cái gì đó như:

def isiterable(obj): 
    try: 
     iter(obj) 
     return True 
    except TypeError: 
     return False 

if isiterable(a): 
    x += sum(similarity(i, j) for i in a for j in b) 

Answer 4

Bạn có thể muốn một cái gì đó như:

if isinstance(a, (dict, list, tuple)): 
    x = sum(similarity(i, j) for i in a for j in b) 
else: 
    x = 0 

Answer 5

Trong một dòng;)

x = sum(similarity(i, j) for i in a for j in b) if isinstance(a, (dict, list, tuple)) else 0 

Answer 6

Bạn có thể sử dụng một số chức năng từ itertools, có lẽ:

from itertools import starmap, product 
x = sum(starmap(similarity, product(a, b))) 

và như những người khác chú ý, cũng có thể vượt qua một tuple các loại để isinstance() nếu bạn thực sự cần phải kiểm tra.