我想我已经明白了单行 for 循环的想法，但现在我遇到了问题。我知道我可以使用以下方式定义数据框列：
df = pd.DataFrame(columns=["columnA"])
list = [0, 1, 2, 3, 4]
df["columnA"] = [i for i in list]
现在我的问题是：是否可以在一行 for 循环中定义 2 列？
我试过这个：
df["columnA"], df["columnB"] = [i, i**2 for i in list]
df["columnA"], df["columnB"] = [[i, i**2] for i in list]
这些都不起作用。我正在使用Python 3.10