Question

我有一个csv文件，如下所示：

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]以删除第一行和最后一行。但这似乎并没有奏效。

任何帮助将不胜感激！

Answer 1

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

with open('thatfile', 'r') as file:
   num_of_lines = file.read().count('\n')
   file.seek(0)
   i = 0
   with open('newfile', 'w') as newfile:
      for line in file:
         if i != 1 and i != num_of_lines:
            newfile.write(line)
         i = i + 1

Python从文件中删除某些行

1 个答案: