Zachowaj spacje przy użyciu split() oraz join() w Pythonie

Question

Mam plik danych z kolumn jak

BBP1 0.000000 -0.150000 2.033000 0.00 -0.150 1.77 

a poszczególne kolumny są oddzielone od siebie różną liczbą białych znaków.

Moim celem jest odczytanie w tych wierszach, wykonanie matematyki w kilku wierszach, na przykład pomnożenie kolumny 4 przez 0,95 i zapisanie ich do nowego pliku. Nowy plik powinien wyglądać jak oryginalny, z wyjątkiem wartości, które zmodyfikowałem.

Moim podejściem byłoby czytanie w wierszach jako pozycje listy. A potem użyłbym split() w tych wierszach, które mnie interesują, co da mi podlistę z indywidualnymi wartościami kolumn. Następnie wykonuję modyfikację, a następnie wpisuję linie z listy do nowego pliku tekstowego.

Problem polega na tym, że mam zmienną liczbę białych znaków. Nie wiem, jak wprowadzić je z powrotem w ten sam sposób, w jaki je przeczytałem. Jedyny sposób, jaki mogłem wymyślić, to zliczyć postacie w linii, zanim je podzielę, co byłoby bardzo uciążliwe. Czy ktoś ma lepszy pomysł na rozwiązanie tego problemu?

Answer 1

Chcesz użyć re.split() w tym przypadku z grupy:

re.split(r'(\s+)', line) 

wrócą obie kolumny i spacje, dzięki czemu można dołączyć linię później z taką samą ilością białych znaków zawartych.

Przykład:

>>> re.split(r'(\s+)', line) 
['BBP1', ' ', '0.000000', ' ', '-0.150000', ' ', '2.033000', ' ', '0.00', ' ', '-0.150', ' ', '1.77'] 

Prawdopodobnie nie chcesz usunąć przełamane od końca.

Answer 2

Inny sposób to zrobić:

s = 'BBP1 0.000000 -0.150000 2.033000 0.00 -0.150 1.77' 
s.split(' ') 
>>> ['BBP1', '', '', '0.000000', '', '-0.150000', '', '', '', '2.033000', '', '0.00', '-0.150', '', '', '1.77'] 

Jeżeli określamy argumentu spacja w podzielonym funkcji, tworzy listy bez jedzenia kolejne spacje. Tak więc oryginalne liczby znaków spacji są przywracane po funkcji "dołącz".