数字+文字列データの文字を削除して数値データにするには

Question

実現したいこと

サンプルが100以上あるデータに年数を表す変数があります。
year
3 years
4 years
5 years
6 yeras
.
.
.
20 years
となっています。"year"をyearを削除して数値データに置き換えたいです。

次のコードを実行すると、2020と返ってきますので、このコードを使おうとしました。

import re
s = "2020年"
result = re.sub(r"\D", "", s)
print(result)

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

年数のデータは"term"に収められています。

import re
s = df_train["term"]
result = re.sub(r"\D", "", s)
print(result)

を実行したら

TypeError: expected string or bytes-like object

と出ました。

どのようなコード書けばそれぞれのデータのyearが消されて数値データになるのでしょうか。

Answer 1

melianさん、ご回答ありがとうございます。実行できました！！

Answer 2

"year" を削除して数値データに置き換えたい

import pandas as pd

df = pd.DataFrame({
    'term': ['3 years', '4 years', '5 years', '6 years', '7 years']
})

#
df['term'] = df['term'].str.extract('^(\d+)').astype(int)
print(df)
print(df.dtypes)

#
   term
0     3
1     4
2     5
3     6
4     7

term    int64
dtype: object