基于数据帧中的两列创建父子树字典

Question

建议我有一个包含国家和城市的数据帧，如下所示：

data = {'Parent':['Netherlands','Belgium','Germany','France'],'Child':['Amsterdam','Brussels','Berlin', '']}

我想创建一本树字典来描述哪个城市属于哪个国家。在这个例子中，我的国家法国没有城市，我不想在字典中有空的值作为子节点。

有人能为我指明正确的方向吗？用什么来达成这个解决方案？

问候

Answer 1

使用熊猫：

df = pd.DataFrame(data)
df.transpose()

---

0         1        2       3
Parent  Netherlands   Belgium  Germany  France
Child     Amsterdam  Brussels   Berlin

或者使用zip:

dict(zip(*data.values()))

---

{'Netherlands': 'Amsterdam', 'Belgium': 'Brussels', 'Germany': 'Berlin', 'France': ''}