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

Question

我使用 python 3.8.5 和 pandas 版本 1.2.1 创建了一个简单的 pandas 数据框

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

我想使用 assign 函数将 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']) ), 这不是你想要的。

如果你想在 lambda 中使用 .assign，你可以这样做:

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