从数据框架向字典添加唯一项

Question

我试图从一个长度为1200的球队名称数据框列中制作一个包含30支NBA球队的字典。我只希望每个团队在字典中出现一次，并与0-29之间的整数配对。我一直很难迭代，到目前为止，我只将数据帧中的第一个值添加到字典中

df1 = df[['team_1']]

teams = dict()

index = 0

for key in df1:

    if key in teams:

        continue
    else:

        teams[key] = index
        index = index + 1

Answer 1

假设df是熊猫数据帧，您可以使用^{}：

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