Перезаписать конкретную строку в текстовом файле с помощью Python - PullRequest
Новая
       12

Перезаписать конкретную строку в текстовом файле с помощью Python

0 голосов
/ 21 декабря 2018

У меня есть разнородный текстовый файл, который мне нужно перезаписать определенную строку в середине его.Я пытался сделать это с помощью функции панд на , но она просто перезаписывает фрейм данных.С CSV у меня тоже не получилось.

Идея состоит в том, чтобы позволить пользователю установить некоторые параметры и затем сохранить это: 

import pandas as pd

file = r'\teste.txt'
col_names = ['Name', 'Elev.', 'MaxDepth', 'InitDepth', 'Shape', 'Curve Name/Params', 'Fevap', 'Psi', 'Ksat', 'IMD']
df = pd.read_csv(file, skiprows=54, skipfooter=79, header=None, names=col_names, delim_whitespace=True, index_col=0)
value1 = df.loc['J1']['Curve Name/Params']  
value2 = df.loc['J2']['Curve Name/Params']  
value3 = df.loc['J3']['Curve Name/Params']  
value4 = df.loc['J4']['Curve Name/Params']

df.at['J1', 'Curve Name/Params'] = input("Insert new area J1:")
df.at['J2', 'Curve Name/Params'] = input("Insert new area J2:")  
df.at['J3', 'Curve Name/Params'] = input("Insert new area J3:")  
df.at['J4', 'Curve Name/Params'] = input("Insert new area J4:")  
print(df[['Curve Name/Params']])
#save the parameters at the txt file

TXT-файл является записью в гидрологической модели (SWMM), и это выглядит так (параметр, который я хотел бы перезаписать, помечен 4 звездочками ****): 

[TITLE]
";;Project Title/Notes"

[OPTIONS]
";;Option             Value"
FLOW_UNITS           CMS
INFILTRATION         CURVE_NUMBER
FLOW_ROUTING         DYNWAVE
LINK_OFFSETS         DEPTH
MIN_SLOPE            0
ALLOW_PONDING        NO
SKIP_STEADY_STATE    NO

START_DATE           07/23/2018
START_TIME           00:00:00
REPORT_START_DATE    07/23/2018
REPORT_START_TIME    00:00:00
END_DATE             07/23/2018
END_TIME             06:00:00
SWEEP_START          01/01
SWEEP_END            12/31
DRY_DAYS             0
REPORT_STEP          00:01:00
WET_STEP             00:05:00
DRY_STEP             01:00:00
ROUTING_STEP         0:00:30 

INERTIAL_DAMPING     PARTIAL
NORMAL_FLOW_LIMITED  BOTH
FORCE_MAIN_EQUATION  H-W
VARIABLE_STEP        0.75
LENGTHENING_STEP     0
MIN_SURFAREA         1.14
MAX_TRIALS           8
HEAD_TOLERANCE       0.0015
SYS_FLOW_TOL         5
LAT_FLOW_TOL         5
MINIMUM_STEP         0.5
THREADS              1

[EVAPORATION]
";;Data Source    Parameters"
";;-------------- ----------------"
CONSTANT         0.0
DRY_ONLY         NO

[OUTFALLS]
";;Name           Elevation  Type       Stage Data       Gated    Route To        "
";;-------------- ---------- ---------- ---------------- -------- ----------------"
OF1              996.1      FREE                        NO                       

[STORAGE]
";;Name           Elev.    MaxDepth   InitDepth  Shape      Curve Name/Params            N/A      Fevap    Psi      Ksat     IMD     "
";;-------------- -------- ---------- ----------- ---------- ---------------------------- -------- --------          -------- --------"
J2               999.1    1.5        0          FUNCTIONAL **10000**      0         0        0        0       
J1               1001     1.5        0          FUNCTIONAL **10000**      0         0        0        0       
J4               997.1    1.5        0          FUNCTIONAL **10000**      0         0        0        0       
J3               999      1.5        0          FUNCTIONAL **10000**      0         0        0        0       

[CONDUITS]
";;Name           From Node        To Node          Length     Roughness  InOffset   OutOffset  InitFlow   MaxFlow   "
";;-------------- ---------------- ---------------- ---------- ---------- ---------- ---------- ---------- ----------"
C1               J1               J3               100        0.013      0          1          0          0         
C2               J2               J4               100        0.013      0          1          0          0         
C3               J3               J4               100        0.013      0          0.9        0          0         
C4               J4               OF1              100        0.013      0          0          0          0         

[XSECTIONS]
";;Link           Shape        Geom1            Geom2      Geom3      Geom4      Barrels    Culvert   "
";;-------------- ------------ ---------------- ---------- ---------- ---------- ---------- ----------"
C1               CIRCULAR     0.5              0          0          0          1                    
C2               CIRCULAR     0.5              0          0          0          1                    
C3               CIRCULAR     0.6              0          0          0          1                    
C4               CIRCULAR     0.7              0          0          0          1                    

[INFLOWS]
";;Node           Constituent      Time Series      Type     Mfactor  Sfactor  Baseline Pattern"
";;-------------- ---------------- ---------------- -------- -------- -------- -------- --------"
J2               FLOW             TR20ANOS         FLOW     1.0      1.0              
J1               FLOW             TR20ANOS         FLOW     1.0      1.0              
J4               FLOW             TR20ANOS         FLOW     1.0      1.0              
J3               FLOW             TR20ANOS         FLOW     1.0      1.0              

[TIMESERIES]
";;Name           Date       Time       Value     "
";;-------------- ---------- ---------- ----------"
TR20ANOS                    0:00       0         
TR20ANOS                    0:01       0.063     
TR20ANOS                    0:02       0.126     
TR20ANOS                    0:03       0.190     
TR20ANOS                    0:04       0.253     
TR20ANOS                    0:05       0.316     
TR20ANOS                    0:06       0.379     
TR20ANOS                    0:07       0.358     
TR20ANOS                    0:08       0.337     
TR20ANOS                    0:09       0.316     
TR20ANOS                    0:10       0.295     
TR20ANOS                    0:11       0.274     
TR20ANOS                    0:12       0.253     
TR20ANOS                    0:13       0.232     
TR20ANOS                    0:14       0.211     
TR20ANOS                    0:15       0.190     
TR20ANOS                    0:16       0.168     
TR20ANOS                    0:17       0.147     
TR20ANOS                    0:18       0.126     
TR20ANOS                    0:19       0.105     
TR20ANOS                    0:20       0.084     
TR20ANOS                    0:21       0.063     
TR20ANOS                    0:22       0.042     
TR20ANOS                    0:23       0.021     
TR20ANOS                    0:24       0.000     

[REPORT]
";;Reporting Options"
INPUT      NO
CONTROLS   NO
SUBCATCHMENTS ALL
NODES ALL
LINKS ALL

[TAGS]

[MAP]
DIMENSIONS 0.000 0.000 10000.000 10000.000
Units      None

[COORDINATES]
";;Node           X-Coord            Y-Coord           "
";;-------------- ------------------ ------------------"
OF1              100.000            0.000             
J2               0.000              100.000           
J1               0.000              200.000           
J4               100.000            100.000           
J3               100.000            200.000           

[VERTICES]
";;Link           X-Coord            Y-Coord           "
";;-------------- ------------------ ------------------"

Заранее благодарен за любую помощь.

Ответы [ 2 ]

0 голосов
/ 21 декабря 2018

Вы можете использовать простую регистрацию здесь.Вам просто нужно:

  • Открыть файл и прочитать строки
  • Получить правильную строку с помощью регулярных выражений для проверки звездочек или того, что вы хотите сопоставить
  • Выполнить операции и перезаписатьпараметры соответственно
  • Записать в файл

Код

Я использовал данные, которые вы указали в data.txt, и сохранил результаты в new.txt.Ниже приведен код, который я использовал. 

import re

with open('data.txt', 'r') as infile,  open('new.txt', 'a') as outfile:
    for curr_line in infile:
        matchObj = re.match(r'[^*]+\*\*[^*]+\*\*[^*]+', curr_line, re.M | re.I)
        if matchObj:
            ## Perform some stuff and overwrite the row
            curr_line = '   Updated Row   '
        outfile.write(curr_line)

Результат

Вот вывод, который я получил 

[TITLE]
";;Project Title/Notes"

[OPTIONS]
";;Option             Value"
FLOW_UNITS           CMS
INFILTRATION         CURVE_NUMBER
FLOW_ROUTING         DYNWAVE
LINK_OFFSETS         DEPTH
MIN_SLOPE            0
ALLOW_PONDING        NO
SKIP_STEADY_STATE    NO

START_DATE           07/23/2018
START_TIME           00:00:00
REPORT_START_DATE    07/23/2018
REPORT_START_TIME    00:00:00
END_DATE             07/23/2018
END_TIME             06:00:00
SWEEP_START          01/01
SWEEP_END            12/31
DRY_DAYS             0
REPORT_STEP          00:01:00
WET_STEP             00:05:00
DRY_STEP             01:00:00
ROUTING_STEP         0:00:30

INERTIAL_DAMPING     PARTIAL
NORMAL_FLOW_LIMITED  BOTH
FORCE_MAIN_EQUATION  H-W
VARIABLE_STEP        0.75
LENGTHENING_STEP     0
MIN_SURFAREA         1.14
MAX_TRIALS           8
HEAD_TOLERANCE       0.0015
SYS_FLOW_TOL         5
LAT_FLOW_TOL         5
MINIMUM_STEP         0.5
THREADS              1

[EVAPORATION]
";;Data Source    Parameters"
";;-------------- ----------------"
CONSTANT         0.0
DRY_ONLY         NO

[OUTFALLS]
";;Name           Elevation  Type       Stage Data       Gated    Route To        "
";;-------------- ---------- ---------- ---------------- -------- ----------------"
OF1              996.1      FREE                        NO

[STORAGE]
";;Name           Elev.    MaxDepth   InitDepth  Shape      Curve Name/Params            N/A      Fevap    Psi      Ksat     IMD     "
";;-------------- -------- ---------- ----------- ---------- ---------------------------- -------- --------          -------- --------"
   Updated Row      Updated Row      Updated Row      Updated Row   
[CONDUITS]
";;Name           From Node        To Node          Length     Roughness  InOffset   OutOffset  InitFlow   MaxFlow   "
";;-------------- ---------------- ---------------- ---------- ---------- ---------- ---------- ---------- ----------"
C1               J1               J3               100        0.013      0          1          0          0
C2               J2               J4               100        0.013      0          1          0          0
C3               J3               J4               100        0.013      0          0.9        0          0
C4               J4               OF1              100        0.013      0          0          0          0

[XSECTIONS]
";;Link           Shape        Geom1            Geom2      Geom3      Geom4      Barrels    Culvert   "
";;-------------- ------------ ---------------- ---------- ---------- ---------- ---------- ----------"
C1               CIRCULAR     0.5              0          0          0          1
C2               CIRCULAR     0.5              0          0          0          1
C3               CIRCULAR     0.6              0          0          0          1
C4               CIRCULAR     0.7              0          0          0          1

[INFLOWS]
";;Node           Constituent      Time Series      Type     Mfactor  Sfactor  Baseline Pattern"
";;-------------- ---------------- ---------------- -------- -------- -------- -------- --------"
J2               FLOW             TR20ANOS         FLOW     1.0      1.0
J1               FLOW             TR20ANOS         FLOW     1.0      1.0
J4               FLOW             TR20ANOS         FLOW     1.0      1.0
J3               FLOW             TR20ANOS         FLOW     1.0      1.0

[TIMESERIES]
";;Name           Date       Time       Value     "
";;-------------- ---------- ---------- ----------"
TR20ANOS                    0:00       0
TR20ANOS                    0:01       0.063
TR20ANOS                    0:02       0.126
TR20ANOS                    0:03       0.190
TR20ANOS                    0:04       0.253
TR20ANOS                    0:05       0.316
TR20ANOS                    0:06       0.379
TR20ANOS                    0:07       0.358
TR20ANOS                    0:08       0.337
TR20ANOS                    0:09       0.316
TR20ANOS                    0:10       0.295
TR20ANOS                    0:11       0.274
TR20ANOS                    0:12       0.253
TR20ANOS                    0:13       0.232
TR20ANOS                    0:14       0.211
TR20ANOS                    0:15       0.190
TR20ANOS                    0:16       0.168
TR20ANOS                    0:17       0.147
TR20ANOS                    0:18       0.126
TR20ANOS                    0:19       0.105
TR20ANOS                    0:20       0.084
TR20ANOS                    0:21       0.063
TR20ANOS                    0:22       0.042
TR20ANOS                    0:23       0.021
TR20ANOS                    0:24       0.000

[REPORT]
";;Reporting Options"
INPUT      NO
CONTROLS   NO
SUBCATCHMENTS ALL
NODES ALL
LINKS ALL

[TAGS]

[MAP]
DIMENSIONS 0.000 0.000 10000.000 10000.000
Units      None

[COORDINATES]
";;Node           X-Coord            Y-Coord           "
";;-------------- ------------------ ------------------"
OF1              100.000            0.000
J2               0.000              100.000
J1               0.000              200.000
J4               100.000            100.000
J3               100.000            200.000

[VERTICES]
";;Link           X-Coord            Y-Coord           "
";;-------------- ------------------ ------------------"
0 голосов
/ 21 декабря 2018

Может быть, вам подойдет простой подход? 

with open('test.txt') as inp:
  with open('result.txt') as outp:
    for line in inp:
      if line:  # skip processing empty lines
        items = line.split()
        if items[0] == 'J1':
          items[123] = input('blahblah')
        line = ' '.join(items)
      outp.write(line)

Я попытался сделать код простым, единственная проблема, которую я вижу, состоит в том, что он не сохраняет количество пробелов в строках после редактирования.Если это имеет значение, код должен быть обновлен.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.

Похожие темы

...