将奇数和偶数分开输出到不同文件

Question

我想把一个输入文件里的奇数和偶数分开，然后把它们分别输出到两个不同的文件里，一个文件放奇数，另一个文件放偶数。

我以前从来没有做过这样的事情。我该怎么做呢？是不是需要运行两次才能得到不同的输出文件？

Answer 1

这是你可以这样做的：

import numpy as np

a = np.loadtxt("test.txt")

odd = []
even = []

for ele in a:
    if ele % 2 == 0:
        even.append(ele)
    else:
        odd.append(ele)

首先，你可以用 a = np.loadtxt("test.txt") 来读取一个文件，然后把里面的内容分成两个数组，一个叫 even（偶数），一个叫 odd（奇数）。最后，你可以用 np.savetxt("odd.txt", odd) 和 np.savetxt("even.txt", even) 来保存这些文件。

编辑：正如 jonrsharpe 提出的，你可以更高效地这样做：

import numpy as np

a = np.loadtxt("test.txt")

np.savetxt("odd.txt", a[a%2==1])
np.savetxt("even.txt", a[a%2==0])