(具体例)simpleなmodelを作る際に上手いcodeが思いつきません。。。

Question

sample.head() 

# output
　　　team1	team2
0	1107	1112
1	1107	1116
2	1107	1124
3	1107	1125
4	1107	1129

vict_teams.head()

# output
	TeamID	vict_per
0	1181	0.808072
1	1242	0.808783
2	1246	0.768182
3	1314	0.755176
4	1112	0.758557

print(vict_teams.shape,sample_submission.shape)

# output
(366, 2) (11390, 2)

上記の様なsample,vict_teamsというdataframe形式のファイルに対して、
sampleに新たなカラム"pred"を追加したいと考えています。

predの内容は、
0.99 ((team1のnumberにおけるvict_teams["TeamID"]のvict_per) >= (team2のnumberにおけるvict_teams["TeamID"]のvict_per))

0.01 ((team1のnumberにおけるvict_teams["TeamID"]のvict_per) < (team2のnumberにおけるvict_teams["TeamID"]のvict_per))

としたいと思います。
(具体的にはsampleの0行目を採用すると、
vict_teams[vict_teams["TeamID"] == 1107] >=
vict_teams[vict_teams["TeamID"] == 1112]
ならsample["pred"][0] = 0.99
となる感じです。)

上記の様にcodeを書いてみたのですが、数が多すぎる為、上手く機能しない様です。
解決するためのうまいcodeがあれば、ご教授頂けますと幸いです。宜しくお願い致します。

Answer 1

いろいろ方法はありそうですが、vect_teams を**Series **データに変換した上で sample.apply()で三項演算子で求めるのがわかりやすいかな

vect_teams = vect_teams.set_index('TeamID')['vect_per']
sample['pred']=sample.apply(lambda d:0.01 if vect_teams[d.team1]<vect_teams[d.team2] else 0.99, axis=1)

Comments

[pickle]

>magichanさん
ご回答ありがとうございます。
vict_teamsをTeamIDをindexとしてSeries dataに変換して、そことsample["team1"],sample["team2"]を照らし合わせてるのですね。わかりました、ありがとうございます！
今後とも宜しくお願い致します。