2017-05-24 46 views
1

我有2个列表,1是嵌套的。我想用条件将它导出到txt。每个不同的字母都以...开头,即:“Letter,A”,后面跟随其元组的第三个元素,否则打印“,”。存在的条件是,如果该元组的第二个元素是在范围与var(从0至5):元组打印元素如果条件else打印范围内的另一个值

var=5 

letter=['A','B','C','D','E','F','G','H'] 
nested_list=[ 
('A', 1, 0), 
('A', 2, 0), 
('B', 1, 9), 
('B', 3, 9), 
('C', 2, 0), 
('C', 4, 0), 
('C', 5, 0), 
('D', 2, 9), 
('E', 3, 0), 
('F', 3, 9)] 

到目前为止我的代码:

bd="Letter," 

for i in range(0,len(nested_list)-1): 
    if nested_list[i][0]!=nested_list[i+1][0]: 
     bd+="\nLetter,%s,"%(nested_list[i][0]) 
     for j in range(0,var): 
      if nested_list[i][1]==j: 
       bd+="%s,"%nested_list[i][2] 
      else: 
       bd+="," 
    elif nested_list[i][0]==nested_list[i+1][0]: 
     bd+="\n" 
     for j in range(0,var): 
      if nested_list[i+1][1]==j: 
       bd+="%s,"%nested_list[i+1][2] 
      else: 
       bd+="," 

print bd 

电流输出:

Letter,A,,,0,,, 
Letter,A,,,0,,,B,,,,9,, 
Letter,B,,,,9,,C,,,,,0,C,,,,,, 
Letter,C,,,,,, 
Letter,D,,,9,,, 
Letter,E,,,,0,, 

预期输出

Letter,A,0,0,,, 
Letter,B,9,,9,, 
Letter,C,,0,,0,0 
Letter,D,,9,,, 
Letter,E,,,0,, 
Letter,F,,,0,, 

有什么建议吗?

回答

1

如果你的列表总是正确的排序,我觉得这是一个很大的用例为itertools.groupby:

In [14]: from itertools import groupby 

In [15]: from operator import itemgetter 

In [16]: for k, group in groupby(nested_list, itemgetter(0)): 
    ...:  plist = ['']*5 
    ...:  for _, idx, val in group: 
    ...:   plist[idx-1] = str(val) 
    ...:  print("Letter,{},{}".format(k, ','.join(plist))) 
    ...: 
Letter,A,0,0,,, 
Letter,B,9,,9,, 
Letter,C,,0,,0,0 
Letter,D,,9,,, 
Letter,E,,,0,, 
Letter,F,,,9,, 
+0

感谢的人,完美的作品! – gilgameshbk