python：小数点以下の表示桁数を指定したい。

Question

機械学習（LightGBM）の特徴量（変数）の重要度を出力したのですが、整数で表示されてしまいます。
小数点以下の桁数を指定して表示するにはどうすればよいでしょうか？

お詳しい方、ご指導をお願いいたします。

 #特徴量の重要度を表示

f_importance = np.array(gbm.feature_importance()) # 特徴量重要度の算出
df_importance = pd.DataFrame({'変数':df_X.columns, '重要度':f_importance})
df_importance = df_importance.sort_values('重要度', ascending=False) # 降順ソート
df_importance

	変数	重要度
1	sex	4
3	kinsou	4
6	pDOI	4
0	age	2
2	rinsyou	2
7	WPOI	2
8	Tumor	2
4	sinkei	0
5	bunka	0
9	myakukan	0

Answer 1

feature_importanceのパラメータにimportance_type='gain'を指定すると小数点表記が返ってくると思います。

f_importance = np.array(gbm.feature_importance( importance_type='gain' )) # 特徴量重要度の算出

[リファレンス]
https://lightgbm.readthedocs.io/en/latest/pythonapi/lightgbm.Booster.html#lightgbm.Booster.feature_importance

Comments

[python_2019]

ご連絡ありがとうございます。
上手くいきました。

追加でお教えください。
リファレンスを読みますと、「gain」と「split」があるようで、両方試しましたら、異なる値が出力されました。
違いは何でしょうか？

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

[hope_mucci]

重要度を計算する際の指標の違いです。

説明すると長くなるので以下の記事と関連リンクを読み込んでください。
https://aotamasaki.hatenablog.com/entry/bias_in_feature_importances
「特徴量重要度とは？」と「3種の特徴量重要度」
https://qiita.com/3000manJPY/items/ef7495960f472ec14377
「ジニ不純度の意味とCART」あたり