用另一个数据框（组合）中的值填充熊猫数据框

Question

这是一个关于组合和 pandas 的小练习。

我目前有一个 pandas 数据框，例如：

|节点1 |节点2|距离|
|悉尼|堪培拉|100|
|墨尔本|悉尼| 200|
|堪培拉|墨尔本| 150|

然后我创建一个新的数据框，因此列标题和行名是节点的唯一值。

mat = pd.DataFrame(index = np.unique(df.node1), columns = np.unique(df.node1))

node1 或 node2 在这里就足够了，因为在构建数据框时顺序并不重要。

这会产生我想要的但填充了 NaN 值。

如何在不写一些丑陋的循环的情况下用距离列中的值填充垫子？

谢谢

Answer 1

首先将pivot 或set_index 与unstack 一起使用：

mat = df.pivot(index='node1', columns='node2', values='distance')
print (mat)
node2      Canberra  Melbourne  Sydney
node1                                 
Canberra        NaN      150.0     NaN
Melbourne       NaN        NaN   200.0
Sydney        100.0        NaN     NaN

---

mat = df.set_index(['node1', 'node2'])['distance'].unstack()
print (mat)
node2      Canberra  Melbourne  Sydney
node1                                 
Canberra        NaN      150.0     NaN
Melbourne       NaN        NaN   200.0
Sydney        100.0        NaN     NaN

然后是combine_first 和T：

mat = mat.combine_first(mat.T)
print (mat)
           Canberra  Melbourne  Sydney
node1                                 
Canberra        NaN      150.0   100.0
Melbourne     150.0        NaN   200.0
Sydney        100.0      200.0     NaN