使用 lambda 了解 pandas 中的分配行为

Question

我使用 python 3.8.5 和 Z3A43B4F8832915D94022C0EF 版本创建了一个简单的 pandas dataframe。

df = pd.DataFrame({'x' : [1,2], 'y' : [3,4]})

我想使用assign function将2列添加到第三列中。

df.assign(c = lambda d: str(d['x']) + str(d['y']))

我期待看到列 'c' 有 ['13', '24']

有人可以帮我理解这种行为吗？

Answer 1

使用str(d['x'])您将获得pd.Series的字符串表示形式（在您执行print(df['x'])时类似），这不是您想要的。

如果您想将.assign与 lambda 一起使用，您可以执行以下操作：

df = pd.DataFrame({"x": [1, 2], "y": [3, 4]})

print(df.assign(c=lambda d: d["x"].astype(str) + d["y"].astype(str)))

印刷：

   x  y   c
0  1  3  13
1  2  4  24