阅读时使用 np.loadtxt 拆分列

Question

有没有办法使用 np.loadtxt 和转换器参数将一列分成两列？

我的文本文件中的行如下所示：

1 2 A=3;B=4

我想把它读成：

[1,2,3,4]

文件很大，逐行读取会很慢。

我试过了：

parse_col = lambda x: [ float(x.split(';')[0].split('=')[1]), int(x.split(';')[1].split('=')[1]) ]

np.loadtxt('demo.txt',usecols=[0,1,2],comments='#',converters={2:parse_col},dtype=int)

谢谢！

Answer 1

您可以创建一个调用解析器的生成器并将其传递给np.genfromtxt：

import re

import numpy as np

def parser(s):
    for i in re.findall('[a-zA-Z]+', s):
        s = s.replace(i, '')
    return s.replace('=', '').replace(';',' ')

gen = (parser(line) for line in open('demo.txt'))
np.genfromtxt(gen, comments='#', usecols=(0, 1, 2, 3))

请注意，正如@PadraicCunningham 所指出的，我使用re.findall 来识别和替换更通用的模式。