python - 如何获取字典中列表的第一个元素并添加到 Python 中的 Pandas Dataframe 列？

Question

我有这样一本字典:

dict = {"key 1": ["val 1", "val 2"],
        "key 2": ["val 3", "val 4", "val 5"],
        "key 3": ["val 6", "val 7"],
         ...
}

我还有一个 pandas 数据框，其中包含如下所有键:

     key
0    key 1
1    key 2
2    key 3
...

我需要在名为 first_key 的数据框中添加一个新列，它为字典中的每个键获取字典中列表的第一个元素，所以它最终是这样的:

     key      first_key
0    key 1    val 1
1    key 2    val 3
2    key 3    val 6
...

我遇到了一些麻烦......做这样的事情是行不通的:

df['first_key'] = df['key'].map(dict[WHAT HERE][0])

:D

Answer 1

尝试:

dct = {
    "key 1": ["val 1", "val 2"],
    "key 2": ["val 3", "val 4", "val 5"],
    "key 3": ["val 6", "val 7"],
}

df["first_key"] = df["key"].apply(dct.get).str[0]
print(df)

打印:

     key first_key
0  key 1     val 1
1  key 2     val 3
2  key 3     val 6

---

或者:

df["first_key"] = df["key"].map(dct).str[0]