为数据框创建新列

Question

import pandas as pd
import numpy as np

train = pd.read_csv("train.csv")

train["DayOfWeek_Monday"] = train.apply(lambda row: row["DayOfWeek"]=="Monday") 

我想要的是使用现有列“DayOfWeek”添加“DayOfWeek_Monday”、“DayOfWeek_Tuesday”等新列。 如何添加新列？

Answer 1

使用get_dummies和DataFrame.reindex进行正确排序并将最后一个DataFrame.join到原始数据：

df = pd.DataFrame({'DayOfWeek':['Sunday','Monday','Wednesday','Thursday', 'Sunday']})

c = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday']
df1 = pd.get_dummies(df['DayOfWeek'], dtype=bool).reindex(c, axis=1, fill_value=False)
df = df.join(df1.add_prefix('DayOfWeek_'))
print (df)
  DayOfWeek  DayOfWeek_Monday  DayOfWeek_Tuesday  DayOfWeek_Wednesday  \
0     Sunday             False              False                False   
1     Monday              True              False                False   
2  Wednesday             False              False                 True   
3   Thursday             False              False                False   
4     Sunday             False              False                False   

   DayOfWeek_Thursday  DayOfWeek_Friday  DayOfWeek_Saturday  DayOfWeek_Sunday  
0               False             False               False              True  
1               False             False               False             False  
2               False             False               False             False  
3                True             False               False             False  
4               False             False               False              True