如何将新列分配给 Pandas 中的现有 DataFrame

Question

我是熊猫的新手。 我正在尝试将新列添加到我现有的 DataFrame 中，但它没有被分配不知道为什么有人可以解释我缺少什么这是我尝试过的

import pandas as pd
df = pd.DataFrame(data = {"test":["mkt1","mkt2","mkt3"],
                         "test2":["cty1","cty2","cty3"]})
print("Before",df.columns)
df.assign(test3="Hello")
print("After",df.columns)

输出

Before Index(['test', 'test2'], dtype='object')
After Index(['test', 'test2'], dtype='object')

Answer 1

Pandas 的assign方法返回一个带有新列的新修改过的数据框，它不会就地修改它。

import pandas as pd
df = pd.DataFrame(data = {"test":["mkt1","mkt2","mkt3"],
                         "test2":["cty1","cty2","cty3"]})
print("Before",df.columns)
df = df.assign(test3="Hello") # <--- Note the variable reassingment
print("After",df.columns)