python - 如何部分读取文本文件并将其写入另一个文本文件

Question

我是新来的，有一个问题。我对编程没有太多了解，因为我只是初学者，所以我希望得到尽可能简单的答案。当然，我会尽力去理解他们!而且英语不是我的母语。为我糟糕的英语道歉。

我想做的任务

我有 a.txt 包含 100 行数据，描述如下:

import numpy as np
b = np.arange(0.005, 0.05, 0.0001)
c = np.arange(1.5, 2.51, 0.01)

with open('a.txt','w') as f:
    for a in range(1,101):
        f.write('{:<3d} {:<3f} {:<3f}\n'.format(a,b[a-1],c[a-1]))

a.txt 上的数据如下所示:

1   0.005000 1.500000
2   0.005100 1.510000
3   0.005200 1.520000
4   0.005300 1.530000
5   0.005400 1.540000
6   0.005500 1.550000
7   0.005600 1.560000
8   0.005700 1.570000
....
97  0.014600 2.460000
98  0.014700 2.470000
99  0.014800 2.480000
100 0.014900 2.490000

现在，我想仅选择第 1 行到第 10 行的数据并将其写入另一个文本文件 b.txt。我怎样才能做到这一点？

• 为了简单起见，我现在正在处理一个非常小的文件，但我想将来对一个非常大(例如几 GB)的文本文件执行此任务，所以我想知道执行任务的方式也可用于处理大文件。

• 如果有任何我没有显示但有必要的信息，请告诉我。我会尽快添加。

非常感谢您的帮助和您的时间。谢谢。

※感谢所有编辑我的帖子的人。它帮助并将帮助我使我的帖子变得更好。

Answer 1

首先，您只能使用 itertools.islice 获取前 n 行，然后编写这些行:

from itertools import islice

n = 10

with open('a.txt', 'r') as infile, open('b.txt', 'w') as outfile:
    first_lines = islice(infile, n)
    outfile.writelines(first_lines)