如何根据行值组合两个长度不同的熊猫数据帧

debugcn 发表于 Dev

tsk_so

我有以下两个熊猫数据框：

数据帧1：

    user_id       animals    
0         1         'dog'
1         1         'cat'
2         1         'cow'
3         2         'dog'
4         2         'cat'
5         2         'cow'
...

数据帧2：（column_D在此任务中不重要）

    location     column_D
0       'CA'            1
1       'MA'            1
2       'AZ'            1
3       'CT'            1
...

我希望基于＃1和＃2创建一个新的数据框＃3：

数据框＃3：

    user_id       animals       location
0         1         'dog'           'MA'
1         1         'cat'           'MA'
2         1         'cow'           'MA'
3         2         'dog'           'AZ'
4         2         'cat'           'AZ'
5         2         'cow'           'AZ'
...

数据帧＃3的第一和第二列与数据帧＃1相同。对于第三列，我希望根据其user_id和数据帧2中的索引来分配位置。例如，对于数据帧＃3中的第0行，由于其user_id = 1，我将检查索引为= 1的数据帧＃2中的位置，然后将该位置（在此示例中为“ MA”）分配给用户。

我搜索了使用concat，map，merge等功能的示例，但找不到与此案例相似的示例。有没有办法完成这项任务？

非常感谢！

sammywemmy

试试地图：

df["location"] = df1.user_id.map(df2.location)

    user_id animals location
0      1    'dog'   'MA'
1      1    'cat'   'MA'
2      1    'cow'   'MA'
3      2    'dog'   'AZ'
4      2    'cat'   'AZ'
5      2    'cow'   'AZ'

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。