python - 从返回元组的函数创建新的 pandas 数据框列

Question

我有一个返回元组的函数:

def pwrs(x):
   return x*x, x*x*x, x*x*x*x

我想将此函数应用于名为 data 的单列数据框:

+-------+
|    x  |
+-------+
|    1  |
|    2  |
|    3  |
|    4  |
+-------+

并根据函数返回元组返回一个包含新列的新数据框:

+---+------+------+------+
| x |  x^2 |  x^3 |  x^4 |
+---+------+------+------+
| 1 |    1 |    1 |    1 |
| 2 |    4 |    8 |   16 |
| 3 |    9 |   27 |   81 |
| 4 |   16 |   64 |  256 |
+---+------+------+------+

遍历行并应用函数:

for _, row in data.iterrows():
    print(pwrs(row['x']))

不知道如何从这里取得进展......

Answer 1

广义

c = np.arange(1, 5)
pd.DataFrame(df.to_numpy() ** c, df.index, c).add_prefix('x^')

   x^1  x^2  x^3  x^4
0    1    1    1    1
1    2    4    8   16
2    3    9   27   81
3    4   16   64  256