在Python中複製列表的一部分

Question

我已經搜索了關於如何做到這一點的相關線程，但我找不到任何東西。

我有一個數組：

x = [a,a,a,b,a,a] 

我想數組的元素複製到一個新的數組，直到我找到「B」。我試圖用循環做到這一點，但我得到一個錯誤，「y沒有定義」，我試圖初始化y，但也沒有工作。有任何想法嗎？我相信有更好的方法來做到這一點。

for ii in x: 
    if x[ii].find(num) == 0: 
     break 
    else: 
     y[ii] = x[ii] 

Answer 1

y = [] 
for e in x: 
    if e == 2: 
     break 
    y.append(e) 

？

Answer 2

你可以使用itertools.takewhile：

>>> x = [1,1,1,2,1,1] 
>>> import itertools 
>>> y = list(itertools.takewhile(lambda i: i != 2, x)) 
>>> y 
[1, 1, 1] 

當使用一個循環，你必須使用y.append;如果你做y[ii] = ...，那麼當你嘗試設置例如你將會得到一個IndexError。具有零元素的數組的第一個元素。此外，當你循環這樣for ii in x:然後ii已經從x元素，即你不要做x[ii]。在你的情況下，這並沒有例外，因爲x[1]將是x的有效元素，但它不會是你所期望的。

Answer 3

試試這個：

x = [1,1,1,2,1,1] 
b = 2 

try: 
    y = x[:x.index(b)] 
except ValueError: 
    y = x[:] 

例如：

In [10]: x = [1,1,1,2,1,1] 
    ...: b = 2 
    ...: 
    ...: try: 
    ...:  y = x[:x.index(b)] 
    ...: except ValueError: 
    ...:  # b was not found in x. Just copy the whole thing. 
    ...:  y = x[:] 
    ...: 

In [11]: y 
Out[11]: [1, 1, 1] 

見list.index()和shallow-copy slice以獲取更多信息。

Answer 4

的方式來獲得與generator expression做的事情：

x = ['a', 'a', 'a', 'b', 'a', 'a'] 
items = list(next(iter([])) if item == 'b' else item for item in x) 
print items 
['a', 'a', 'a'] 

Answer 5

雖然它在某種程度上類似於威爾的答案，但在這裏它說明了如何使用slice內置切片對象：

>>> x = ['a','a','a','b','a','a'] 
>>> s = slice(x.index('b')) 
>>> 
>>> x[s] 
['a', 'a', 'a'] 

Answer 6

我會按b索引分割數組：

>>> x = ['a','a','a','b','a','a'] 
>>> c = x.index('b') 
>>> x[:c] 
['a', 'a', 'a'] 
>>> x[c:] 
['b', 'a', 'a']