Используйте 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)) 

Обратите внимание, что я использовал re.findall, чтобы определить и заменить более общую картину, как отметил @PadraicCunningham.