如何將新列分配給 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)