python - 如何分配将元组抛出到数据帧的两个不同列的函数

Question

我定义了以下函数，它生成一个元组作为输出，并且我需要将结果分配给数据帧的两个不同列。

这些函数只是获取 x 和 y 坐标并将其转换为不同的坐标:

plot_shot(x,y)

我开发了以下有效的代码，但我正在寻找一种更智能的解决方案，允许我在 apply lambda 行中分配元组:

df_shots['temporary'] = df_shots.apply (lambda a: plot_shot(a['x'],a['y'],0,15,-3,'left','bottom'), axis=1)
df_shots['x_converted'] = df_shots['temporary'].apply (lambda x: x[0])
df_shots['y_converted'] = df_shots['temporary'].apply (lambda x: x[1])
df_shots.drop('temporary', axis=1, inplace = True)

我假设这可以在一行中完成，而不是这个。

提前致谢。

Answer 1

我还没有测试过代码，但也许下面的一些代码可以工作。

这可能行不通:

df_shots['x_converted'], df_shots['y_converted'] = df_shots.apply (lambda a: plot_shot(a['x'],a['y'], ...), axis=1)

这可能是:

df_shots['x_converted'] = df_shots.apply (lambda a: plot_shot(a['x'],a['y'], ...)[0], axis=1)
df_shots['y_converted'] = df_shots.apply (lambda a: plot_shot(a['x'],a['y'], ...)[1], axis=1)

抱歉，没有要测试的 pandas 安装..