python - 如何解析python输入文件

Question

我有一个包含制表符分隔值的文件，例如:

"1" "12345" "abc" "def"
"2" "67890" "abc" "ghi"
"3" "13578" "jkl" "mno"

我不知道如何从输入文件中获取任意数字，如果前 5 位数字与输入文件第二列中的内容匹配，那么该行中的所有内容都将导出到另一个文件中。

例如:输入文件:“67890123”

output file: "2"   "67890"   "abc"   "ghi"

Answer 1

您可以使用pandas 包来读写您的数据文件。

from __future__ import with_statement
import pandas as pd

inputFileName = "D:/tmp/inputfile.txt"
dataFileName = "D:/tmp/data.csv"
outputFileName = "D:/tmp/outputfile.txt"

data = pd.read_csv(dataFileName, sep=' ', header=None)

with open(inputFileName) as f:
    input = f.readlines()
input = [int(x[0:5]) for x in input]

output = pd.DataFrame()
for value in input:
    output = output.append(data[data[data.columns[1]] == value])

output.to_csv(outputFileName, sep=' ', header=None, index=False)

所以如果你的输入文件有

67890123
13578010

你的数据是

"1" "12345" "abc" "def"
"2" "67890" "abc" "ghi"
"3" "13578" "jkl" "mno"

输出文件为:

2 67890 abc ghi
3 13578 jkl mno