7

我正在合并两个数据框，merge(..., how='left')因为我只想保留与“左”数据框匹配的条目。问题是合并操作似乎删除了我最左边的数据框的索引，如下所示：

import pandas
df1 = pandas.DataFrame([{"id": 1,
                         "name": "bob"},
                        {"id": 10,
                         "name": "sally"}])
df1 = df1.set_index("id")
df2 = pandas.DataFrame([{"name": "bob",
                         "age": 10},
                        {"name": "sally",
                         "age": 11}])

print "df1 premerge: "
print df1
df1 = df1.merge(df2, on=["name"],
                how="left")
print "merged: "
print df1
# This is not "id"
print df1.index
# And there's no "id" field
assert ("id" in df1.columns) == False

合并前，df1被id. 合并操作后，合并数据框只有默认数字索引，并且该id字段被删除。如何进行这种合并操作但保留最左侧数据框的索引？

澄清一下：我希望将 的所有列df2添加到df1具有匹配id值的每个条目中。如果 in 中的条目df2的id值不在 中df1，则不应将其合并到中（因此是how='left'）。

编辑：我可以作为一个黑客做：df1.reset_index()但合并然后再次设置索引，但如果可能的话我不喜欢，似乎合并不应该删除索引。谢谢。

4

1 回答 1

4

您已经指出在合并之前执行 reset_index 并在之后执行 set_index，这是可行的。我知道在合并中保留索引的唯一方法是让合并在至少一个正在合并的数据帧上包含一个索引。所以在这里，你可以这样做：

In [403]: df2 = df2.set_index('name')

In [404]: df1.merge(df2, left_on='name', right_index=True)
Out[404]: 
     name  age
id            
1     bob   10
10  sally   11

将 df2 的索引（我们从其“名称”列中获取）与 df1 上的“名称”列合并。

这是有道理的，因为否则生成的数据帧的索引是不明确的，因为它可能来自任一数据帧。

于 2013-05-27T18:19:05.813 回答