pythonでのテキストファイルからのインプット

Question

前提・実現したいこと

python初心者です
pythonでtextファイルからのインプットを考えています
"[xxx]"で始まる箇所をsectionとしてsection毎にリストでアウトプットしたいと思っています。

発生している問題・エラーメッセージ

flugを用いて，処理を分岐させていますが，シンプルなコードにできませんか？

該当のソースコード

input.py

def input(file_input):
    lines = [line.rstrip() for line in open(file_input)]
    while "" in lines:
        lines.remove("")
    flug = 0
    title = []
    parameter = []
    node = []
    element = []
    for line in lines:
        line = line.lower()
        if line[0:1] == "[":
            flug = 0
        if line == "[title]":
            flug = 1
        if line == "[parameter]":
            flug = 2
        if line == "[node]":
            flug = 3
        if line == "[element]":
            flug = 4
        if flug == 1:
            title.append(line)
        if flug == 2:
            parameter.append(line)
        if flug == 3:
            node.append(line)
        if flug == 4:
            element.append(line)
    print(title)
    print(parameter)
    print(node)
    print(element)

if __name__ == "__main__":
    input("test.inp")

------------------------------------
test.inp
[Title]
test input
2018.02.07

[Parameter]
ndim=4
ncal = 5

[Node]
test.nod

[Element]
test.ele


[End]

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

セクションをキーとしたデータのリストを辞書で保持する例です。

import re

# データ読込、空行の除去などなど

lines = ['[hoge]','a','b','[huga]','c','d']
dic = {}
sec = ''
for line in lines:
    m = re.match(r"[(\w+)]", line)
    if m:
        sec = m.group(1)
        dic[sec] = []
    else:
        dic[sec].append(line)

print(dic) # {'hoge': ['a', 'b'], 'huga': ['c', 'd']}

Comments

[can110]

かぶった。。

[tera_nsykuh]

ありがとうございます。
非常にシンプルなコードで，勉強になりました。

Answer 2

def get_input(filepath):
    with open(filepath, 'r') as f:
        lines = [l.rstrip() for l in f.readlines()]
    results = {}
    for line in lines:
        if line.startswith('['):
            key = line[1:-1].lower()
            results[key] = []
            tlist = results[key]
        else:
            if len(line) > 0:
                tlist.append(line)
    return results

if __name__ == "__main__":
    results = get_input("test.inp")

    for k,v in results.items():
        print('-'*20)
        print(k)
        for i in v:
            print(i)

Comments

[tera_nsykuh]

ありがとうございます。
実現したい通りのコードです。

Answer 3

解決後ではありますが、正規表現を使った方が綺麗に書ける気がしたので書いてみました。

my_file_load.py

import re
import pprint

def file_load(file_name):
    with open(file_name) as f:
        doc = f.read()
    
    content_dic = {}
    
    elms = re.findall(r'[(\w+)]\n(.*?(?=[))',doc,flags=re.DOTALL)
    
    for elm in elms:
        content_dic[elm[0]] = elm[1].rstrip('\n')
    
    return content_dic

if __name__=='__main__':
    dic = file_load('test.txt')
    pprint.pprint(dic)

test.txt

[Title]
test input
2018.02.07

[Parameter]
ndim=4
ncal = 5

[Node]
test.nod

[Element]
test.ele


[End]

実行結果

{'Element': 'test.ele',
 'Node': 'test.nod',
 'Parameter': 'ndim=4\nncal = 5',
 'Title': 'test input\n2018.02.07'}

それとファイル名や関数名をinputにするのは、pythonには組み込み関数としてinputがあるので、避けた方がいいです。

Comments

[namnium1125]

かぶった。。(大分遅刻)

[tera_nsykuh]

ありがとうございます。
関数名についてのコメントも勉強になりました。