range（）與'禁止'值？

Question

我們來舉個例子吧。我有這樣一個普通的功能：

def function(x): 
    for b in range(0, 5): 
     print(b) 

我想有x作爲例外的範圍;這樣一來，它會做這樣的事情：

def function(x): 
    for b in range(0, x): 
     print(b) 
    for b in range(x+1, 5): 
     print(b) 

但如果我想要做的不僅僅是一個簡單的print()時間越長，就會延長我的功能很多。有沒有解決方案呢？

Answer 1

跳過x內循環：

for b in range(0, 5): 
    if b == x: 
     continue # skip to next iteration 
    print(b) 

Answer 2

兩個子範圍可能是"chain版」在一起：

from itertools import chain 

def function(x): 
    for b in chain(range(0, x), range(x+1, 5)): 
     print(b) 

或者你也可以使用一個generator expression：

def function(x): 
    for b in (v for v in range(0, 5) if v != x): 
     print(b) 

後者可以很容易地推廣到支持e包含多個值：

def function(*x): 
    for b in (v for v in range(0, 5) if v not in set(x)): 
     print(b) 

function(3)  # -> 0 1 2 4 
function(1, 3) # -> 0 2 4