python3 文字列内の文字と数値を分離する

Question

文字列データの中に数値に変換可能な部分がある場合、その数値部分のみを抽出したいのですがどうしたらよいでしょうか？

[入力]

80kg 65kg 72kg 69kg 100kg  #半角の空白を空けて体重データ(kg)を入力します。

[コード]

weight = input().rstrip().split(" ")  #データをリストに格納します。

#ここでデータの数値部分だけを取り出し、最終的に合計値(kg)を出したい。

[出力]

386kg

先頭から2文字分だけ取り出してint()で数値化することも考えたんですが、5つ目のデータのように桁数が異なるデータに対しては使えません。

また単位が別の単位だった場合にも対応できるようにするにはどうしたらよいでしょう？

Answer 1

正規表現が便利です。

import re

srcs = '80kg 65.5kg 72W 69km 100t'.split()
for src in srcs:
    m = re.match(r'^([\d.]+)(\D+)$', src)
    assert m
    
    value, unit = float(m.group(1)), m.group(2)
    print(
        f'{src} -> "{value}", "{unit}"'
    )

実行結果 Wandbox

80kg -> "80.0", "kg"
65.5kg -> "65.5", "kg"
72W -> "72.0", "W"
69km -> "69.0", "km"
100t -> "100.0", "t"

Answer 2

単位の表記が限定されているなら、
"kg" -> " kg" のように " " を insert してしまえば、あとは split() で処理できます。

kg,py

units = ("kg", "W", "km", "t")
srcs = '80kg 65.5kg 72W 69km 100t'

s = srcs
for unit in units:
    s = s.replace(unit, " " + unit)
print(s.split())

n_unit_list = list(zip(*[iter(s.split())] * 2))
print(n_unit_list)

x = [(float(d), u) for (d, u) in zip(*[iter(s.split())] * 2)]
print(x)

実行例

Answer 3

kgを取り除くだけならこれでどうでしょう？

# [入力データ]
#80kg 65kg 72kg 69kg 100kg

input_data = input()

def func(s):
    return int(s.replace('kg',''))

weight = list(map(func, input_data.split(" ")))

print(weight,'リストの内容確認')
print('{}kg'.format(sum(weight)))

#[出力]
#386kg