pandasで読み込んだcsvの単位付きの値を単位無しに変換したい

Question

pandasで読み込んだcsvファイル[df_train]に[距離]という行があって、その値に[km]や[m]といった単位が付いています。

列1	列2	距離
	1.2 km
	1.3 km
	9 m

この時下の表のように、[km]や[m]といった単位を取り除きつつ、[m]単位の値にする前処理の方法を知りたいです。

列1	列2	距離
	1200
	1300
	9

よろしくお願い致します。

Answer 1

この手の（文字列を解析して数値に変換するような）処理は、文字列のフォーマットが一律であればそれほど難しくないのですが、想定しているフォーマット外の文字が紛れ込んだりすると直ぐに動作しなくなるので注意が必用です。

で今回の場合、距離列のフォーマットが全て
"(数値) (単位)"
のように(数値)と(単位)がスペース区切りとなっており、(単位)の種類として 'm'と'km'の２種類しかないという想定であれば

tmp = df['距離'].str.split(expand=True)
df['距離'] = tmp[0].astype('float').mul(tmp[1].map({'m': 1, 'km':1000}))

と書けます。

最初に書いたとおり文字列に想定外（単位が無かったり、スペースが無かったり、スペースが全角だったり・・）の場合は動作しま
せんのでご注意ください

import pandas as pd
df = pd.DataFrame({
    '列1':[0,0,0],
    '列2':[0,0,0],
    '距離':['1.2 km','1.3 km','9 m']})
#    列1  列2      距離
# 0   0   0  1.2 km
# 1   0   0  1.3 km
# 2   0   0     9 m

tmp = df['距離'].str.split(expand=True)
df['距離'] = tmp[0].astype('float').mul(tmp[1].map({'m': 1, 'km':1000}))
# 0   0   0  1200.0
# 1   0   0  1300.0
# 2   0   0     9.0

Answer 2

こんな感じでしょうか。

def func(a):
    # まずは空白を削除
    a = a.replace(" ", "")
    
    # kmかどうかを判定する
    if a.find("km") != -1: # kmなら、1000倍する
        a = a.replace("km", "")
        a = float(a) * 1000
    else: # mなら、そのまま
        a = a.replace("m", "")
        a = float(a)
    a = int(a)
    return a


x = "1.2 km"
y = "9 m"

xx = func(x)
yy = func(y)


print(xx)
print(yy)

pandasで取ってきた文字列をこのfuncに入れればできます！

不明点があれば、何なりとご連絡ください。