python - 如何从另一个文本文件写入本地最大值的文本文件？

Question

我有一个包含两列的文本文件。我想找到第一列的局部最大值。这一步我没有问题。之后，我想在一个新的文本文件中写入这些局部最大值的完整行。我在这一步有问题。我的代码中的问题与“for line in range peaks:”行有关。这里的 peaks 是那些包含局部最大值的行的索引。

f = open ('rr.txt', 'r')
data = f.readlines()
a = [float(line.split()[0]) for line in data]
a = np.array(a)
peaks, _ = find_peaks(a, height=0)

with open('peaks.txt', 'w') as f:
    for line in range peaks:
        f.write(line)

输入：

0 11
3 11
2 12
1 13
4 20
2 21

峰是：

array([1,4])

a[峰]= 3 ,4

和所需的输出：

3 11
4 20

标签： pythontextmax