Pythonの文字列のリストから数値を抽出する

データ操作にPythonを使用しているときに、要素が固定パターンの文字と数字の組み合わせであるリストに出くわす場合があります。この記事では、将来の計算に使用できる文字から数字を分離する方法を説明します。

分割あり

分割関数は、区切り文字として扱われる文字を使用して文字列を分割します。以下のプログラムでは、リスト要素の文字とテキストの間の区切り文字としてハイフンが使用されています。これをforループと一緒に使用して、それぞれをキャプチャします

例

listA = ['Mon-2', 'Wed-8', 'Thu-2', 'Fri-7']

# Given list
print("Given list : " + str(listA))

# Extracting numbers
num_list = [int(i.split('-')[1]) for i in listA]

# print result
print("List only with numbers : ",num_list)

出力

上記のコードを実行すると、次の結果が得られます-

Given list : ['Mon-2', 'Wed-8', 'Thu-2', 'Fri-7']
List only with numbers : [2, 8, 2, 7]

地図とisnumeric付き

このアプローチでは、各要素を調べて、各要素に存在する数値部分を確認します。マップ関数は、各要素に同じ関数を繰り返し適用するために使用されます。

例

listA = ['Mon-2', 'Wed-8', 'Thu-2', 'Fri-7']

# Given list
print("Given list : " + str(listA))

# Extracting numbers
num_list = list(map(lambda sub:int(''.join(
[i for i in sub if i.isnumeric()])), listA))

# print result
print("List only with numbers : ",num_list)

出力

上記のコードを実行すると、次の結果が得られます-

Given list : ['Mon-2', 'Wed-8', 'Thu-2', 'Fri-7']
List only with numbers : [2, 8, 2, 7]