2017-08-14 64 views
0

删除某些线我有一个csv文件,看起来像以下:Python从文件

CCC, rev, pIndex, no. 
23, 2.5, 0, 4 
34, 4.3, 1, 3 
27, 7.9, 2, 5 
12, 5.7, 3, 7 

其中每行中的第三个元素被称为pIndex。我有代码复制每一行,而相应地更新pIndex。因此输出看起来如下:

CCC, rev, pIndex, no. 
23, 2.5, 0, 4 
23, 2.5, 1, 4 
34, 4.3, 2, 3 
34, 4.3, 3, 3 
27, 7.9, 4, 5 
27, 7.9, 5, 5 
12, 5.7, 6, 7 
12, 5.7, 7, 7 

下面是做的代码:

with fileinput.input(inplace=True) as f: 
    n = 0 
    for line in f: 
     line = line.rstrip('\n') 
     numbers = line.split(sep = ',') 
     for i in range(2): 
      numbers[3] = n 
      n += 1 
      s = '' 
      for j in numbers: 
       s += str(j) + ',' 
      s = s.rstrip(',') 
      print(s) 

现在我想删除每一个第一和最后一行的一个重复,所以基本上删除第一线(在标题后,pIndex 0)和文件中的最后一行。我正在使用fileinput并试图将结果写回到同一个文件中。我试过

result = [] 
result.append(s) 

这使得每一行都是一个单独的数组。然后,我会做出另一个数组 result_all,它将包含所有行(作为数组),然后做result_all[1:-1]摆脱第一行和最后一行。但是这似乎并不奏效。

任何帮助将不胜感激!

回答

0

使用可以迭代行并跳过第1行和最后一行,但您需要知道数字行。

+0

谢谢,但我试图使它通用许多文件,所以我想通过命令行像'python script.py filename.csv' –