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

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是Pandas DataFrame，您可以使用dict comprehension：

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