pandas新建dataframe_如何迭代pandas dataframe并创建新列

我认为您可以使用双^{}，什么是更快的循环：df['Column3'] = np.where(df['Column2']==variable1, variable2,

np.where(df['Column2']==variable3, variable4))

如果两个条件都是False，则需要添加变量：

^{pr2}$

样品：df = pd.DataFrame({'Column2':[1,2,4,3]})

print (df)

Column2

0 1

1 2

2 4

3 3

variable1 = 1

variable2 = 2

variable3 = 3

variable4 = 4

variable5 = 5

df['Column3'] = np.where(df['Column2']==variable1, variable2,

np.where(df['Column2']==variable3, variable4, variable5))

print (df)

Column2 Column3

0 1 2

1 2 5

2 4 5

3 3 4

另一个解决方案，谢谢^{}：df['Column4'] = df.Column2.map({variable1: variable2, variable3:variable4}).fillna(variable5)

print (df)

Column2 Column3 Column4

0 1 2 2.0

1 2 5 5.0

2 4 5 5.0

3 3 4 4.0