使用 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