2016-04-29 13 views
-1

私は、チーム名の長さ1200データフレームの列から30のNBAチームからなる辞書を作成しようとしています。私は各チームを辞書に一度出現させ、0から29までの整数にペアリングしたいだけです。私は困難な時間を繰り返してきました。これまでのところ、データフレームの最初の値を辞書に追加するだけです。データフレームから辞書にユニークな項目を追加する

dfはパンダデータフレームである、あなたは dict comprehensionを使用したと仮定すると
df1 = df[['team_1']] 

teams = dict() 

index = 0 

for key in df1: 

    if key in teams: 

     continue 
    else: 

     teams[key] = index 
     index = index + 1 

答えて

3

teams = {team:i for i, team in enumerate(df['team_1'].unique())} 
+0

これは動作しますが、おかげでたくさん! –

関連する問題