在Python中切割一維列表爲兩部分

Question

我有兩個一維列表x和y（未知的巨大長度）取自一個數據表。

我想取值x < 5和相應的y值到列表的一個set，說{x1, y1}。

另外我想x > 6的值和相應的y值到列表的另一個set，說{x2, y2}。

我該怎麼辦？

Answer 1

這裏有一個選項

z = zip(x,y) #z = [(x[0],y[0]), ...] 

smaller = [(xval,yval) for (xval,yval) in z if xval<= 5] #those with x<=5 
larger = [(xval,yval) for (xval,yval) in z if xval>=6] #those with x>=6 

x1,y1 = map(list,zip(*smaller)) #break them apart and make the results lists, not tuples. 

x2,y2 = map(list,zip(*larger)) 

這是一種很酷如何zip(*list)撤銷list = zip(.,.)。請注意，如果x=5.5是一個選項，則此代碼中存在一個錯誤。最好把它看作xval> 5的大型案件，但我沒有這樣做，因爲從你的問題不清楚5.5的標準應該是什麼。