我有以下两个熊猫数据框:
数据帧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等功能的示例,但找不到与此案例相似的示例。有没有办法完成这项任务?
非常感谢!
试试地图:
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] 删除。
我来说两句