pandas是一個(gè)用于數(shù)據(jù)處理和分析的Python庫(kù)，它提供了豐富的功能來(lái)操作數(shù)據(jù)。
在pandas中，我們可以使用多種方法為DataFrame新增一列。本文介紹10種不同的操作來(lái)為pandas DataFrame新增一列：
import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 直接賦值法
df['C'] = [7, 8, 9]
df
import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 使用assign方法
df = df.assign(C=[7, 8, 9])
df
import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 使用insert方法
df.insert(2, 'C', [7, 8, 9])
print(df)
insert方法可以在指定位置插入新列。
import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 使用join方法
new_column = pd.Series([7, 8, 9], name='C')
df = df.join(new_column)
print(df)
import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 使用concat方法
new_column = pd.DataFrame({'C': [7, 8, 9]})
df = pd.concat([df, new_column], axis=1)
print(df)
import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 使用apply方法
def add_new_column(row):
if row['A']>2:
return row['A']**2 + row['B']*100
else:
return row['A']-row['B']
df['C'] = df.apply(add_new_column, axis=1)
print(df)
本方法是萬(wàn)能方法，經(jīng)常用到，在經(jīng)過(guò)復(fù)雜運(yùn)算并賦值到新的一列的場(chǎng)景使用尤其多。
import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 使用apply方法結(jié)合lambda函數(shù)
df['C'] = df.apply(lambda row: row['A'] * 2, axis=1)
print(df)
import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 使用query方法
df['C'] = df.query('A > 1')['B']
print(df)
import pandas as pd
# 創(chuàng)建一個(gè)DataFrame
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 使用loc方法添加新列'C'
df.loc[:, 'C'] = [7, 8, 9]
print(df)
loc方法用于基于標(biāo)簽選擇數(shù)據(jù)，可以用于添加新列。
import pandas as pd
# 創(chuàng)建兩個(gè)DataFrame
data1 = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df1 = pd.DataFrame(data1)
data2 = {'C': [7, 8, 9]}
df2 = pd.DataFrame(data2)
# 使用merge方法添加新列'C'
df = pd.merge(df1, df2, left_index=True, right_index=True)
print(df)
merge方法可以根據(jù)指定的鍵將兩個(gè)DataFrame合并，用于添加新列。
本文介紹的pandas新增一列常用方法，大家可根據(jù)處理數(shù)據(jù)的實(shí)際場(chǎng)景選擇適合的方法使用。
