テキスト文字列における":"と","で区切ってcsv化したい

Question

前提

data   0: 10,   20
data   1: 20,   30
data   2: 30,   40
data   3: 40,   50
data   4: 50,   60
data   5: 60,   70

の形式で記述されたテキストファイルがあります。
このファイルにおける:と,を目印にして、以下のようなcsvファイルを作成したいです。

試したこと

with open(filename, encoding="shift-jis") as f:
    lines = f.readlines()

はじめに、上記の処理によりテキストを1行ずつリストに格納することを試みました。
その上で、split()によるテキスト分割をしようと思い 試行したところ以下の出力となりました。

data = linlines[0].split()
#['data', '0:', '10,', '20']

data[0]およびdata[1]が分割されてしまってる上に、data[2]に余計な,が含まれる結果となってしまいました。
解決策をご教授いただけるとありがたいです。

Answer 1

import re

filename = 'data.txt'
with open(filename, encoding="shift-jis") as f:
    csv_lines = f.read().translate(str.maketrans({' ': '', ':': ','}))
    #csv_lines = re.sub(r'[ :]', lambda g: {' ': '', ':': ','}[g[0]], f.read())

print(csv_lines)

# data0,10,20
# data1,20,30
# data2,30,40
# data3,40,50
# data4,50,60
# data5,60,70

Comments

[katamalix]

ありがとうございます。
このようにスマートに記述できることに感動しました。
参考にさせていただきます。

[melian]

str.translate() の方が簡便なので、そちらに書き直しました。(import re が不要になります)

[katamalix]

ありがとうございます。

質問ではtxtファイルを読み込むことを想定した記述としましたが、
実際は膨大なtxtファイルから該当部分を抽出したオブジェクトを操作しております。
すなわち、
with open(filename, encoding="shift-jis") as f:
lines = f.readlines()

lines_len = len(lines)
line_arr = np.array(lines)
smp_start = np.where(line_arr=="START")[0][-1]
smp_end = np.where(line_arr=="END")[0][-1]
smp_data = lines[smp_start:smp_end]
のような操作をしており、smp_data (リスト) が手元にある状態です。
このリストを一旦txt出力し、再度with文で読ませればmelian様の記述した文で対応できるのですが、
txt出力を介さない方法があればご教授頂きたいです。

[katamalix]

for i in range(len(smp_data)):
smp_data[i] = re.split( ",|:",smp_data[i].translate(str.maketrans({' ': '', ':': ',', '\n': ''})))

この記述とし2次元配列化、to_csv()で実現できました。ありがとうございました。

Answer 2

以下のような手順はどうでしょうか。

1. 行からすべての空白を削除。
2. 「:」 で分割する。
3. 2つめを「,」で分割する。

re.split()を使えば「:もしくは,で分割する」もできますね。

Comments

[katamalix]

なるほど、はじめに空白を詰めてしまえば「空白で分割」したときに","が残ることもないですね。ありがとうございます！