直観でわかる、Pandas → Scikit-learnへの渡し方 - Qiita

何があったか 手元に Pandas の DataFrame がある条件で、ひさしぶりに Scikit-learn を使おうとしたら、どういう形で渡すんだっけ？と少し迷いました。同じ境遇のとき、直観的に思い出せるように書き出します。 前提 手元にDataFrame がある前提です( 今回はダミーデータで DataFrame を自作します )。 直観的理解を目指して、サンプルでは説明変数３つ ( X1, X2, X3 ) だけの、超簡易なものを使います。 目的変数 ( Y ) はバイナリラベル ( 1, 0 ) とします。 import pandas as pd data = [[1,1,0,0], [0,0,0,1], [0,0,1,1]] index = ["a", "b", "c"] columns = ["Y","X1","X2","X3"] df = pd.DataFrame(dat