如何利用Pandas,将一个文件的某列值根据另一个文件对应列名进行重新赋值?

发布于 2022-09-07 11:46:53 字数 380 浏览 13 评论 0

现在有两个表(txt文件),需要将其中一个文件某列值修改成另一个文件对应的列值,具体如下:
文件1
clipboard.png

文件2
clipboard.png

我需要将文件1的x,y列的值根据列名t和userid从文件2获取对应的x,y值进行重新赋值。由于Pandas这块运用不熟悉,希望有知道的指导一下,多谢哈!

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(2

扎心 2022-09-14 11:46:53
import pandas as pd
import numpy as np
df1 = pd.DataFrame([[0, 1, 2, 3], [4, 5, 6, 7]], columns=['t', 'user_id', 'x', 'y'])
>>> df1
   t  user_id  x  y
0  0        1  2  3
1  4        5  6  7
df2 = pd.DataFrame([[0, 1, 0, 0], [4, 5, 0, 0], [5, 6, 7, 8]], columns=['t', 'user_id', 'x', 'y'])
>>> df2
   t  user_id  x  y
0  0        1  0  0
1  4        5  0  0
2  5        6  7  8
def f(row):
    if np.isnan(row['x_y']):
        row['x'], row['y'] = row['x_x'], row['y_x']
    else:
        row['x'], row['y'] = row['x_y'], row['y_y']
    return row

res = pd.merge(df2, df1, how='left', on=['t', 'user_id']).apply(f, axis=1)[['t', 'user_id', 'x', 'y']]
>>> res
     t  user_id    x    y
0  0.0      1.0  2.0  3.0
1  4.0      5.0  6.0  7.0
2  5.0      6.0  7.0  8.0
孤千羽 2022-09-14 11:46:53

为提升性能,应减少 pandas.DataFrame.apply() 的逐行操作,在本例中可改用 numpy.where() 二元操作符,如下

In [1]: import pandas as pd

In [2]: import numpy as np

In [3]: df1 = pd.DataFrame({'t': [1,2,3], 'user_id': [10,20,30], 'v': [1.1,2.2,3.3]})

In [4]: df1
Out[4]: 
   t  user_id    v
0  1       10  1.1
1  2       20  2.2
2  3       30  3.3

In [5]: df2 = pd.DataFrame({'t': [4,1,2], 'user_id': [40,10,20], 'v': [400,100,200]})

In [6]: df2
Out[6]: 
   t  user_id    v
0  4       40  400
1  1       10  100
2  2       20  200

In [7]: df3 = pd.merge(df1, df2, how='right', on=['t', 'user_id'])

In [8]: df3
Out[8]: 
   t  user_id  v_x  v_y
0  1       10  1.1  100
1  2       20  2.2  200
2  4       40  NaN  400

In [9]: df3['v'] = np.where(np.isnan(df3.v_x), df3.v_y, df3.v_x)

In [10]: df3
Out[10]: 
   t  user_id  v_x  v_y      v
0  1       10  1.1  100    1.1
1  2       20  2.2  200    2.2
2  4       40  NaN  400  400.0

In [11]: del df3['v_x']

In [12]: del df3['v_y']

In [13]: df3
Out[13]: 
   t  user_id      v
0  1       10    1.1
1  2       20    2.2
2  4       40  400.0
~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文