Python列表中的条件

时间:2012-09-07 14:50:23

标签: python

目前我的循环中有一个新列表,大部分内容都会创建一个新列表,其中包含数据中原始列表x的几乎所有元素。但是,我正在创建一个条件,只有在条件为x [5]时才会增加xnew的特定索引。有更清洁的方法来执行以下操作吗?:

for x in data:
  xnew = [x[1],x[2],x[3],x[4],0,0,0]
  if(x[5]==2):
    xnew[4] = x[8] 
  elif(x[5]==9):
    xnew[5] = x[8] 
  else:
    xnew[6] = x[8]

1 个答案:

答案 0 :(得分:8)

xmap = {2: 4, 9: 5}

 ...

xnew[xmap.get(x[5], 6)] = x[8]