![](/img/trans.png)
[英]Creating a new column based on if-elif-else condition in python pandas
[英]Error while creating a new column based on if-elif-else condition
我正在尝试创建一个新列并将其添加到 dataframe 的周。此外,我正在根据 if else 条件执行此操作。 我正在使用下面的代码。
# List of the conditions
conditions = [
(df_sup[(df_sup['Date']>='2016-11-14') & (df_sup['Date']<'2016-11-21')]),
(df_sup[(df_sup['Date']>='2016-11-21') & (df_sup['Date']<'2016-11-28')]),
(df_sup[(df_sup['Date']>='2016-11-28') & (df_sup['Date']<'2016-12-05')]),
(df_sup[(df_sup['Date']>='2016-12-05') & (df_sup['Date']<'2016-12-12')]),
(df_sup[(df_sup['Date']>='2016-12-12') & (df_sup['Date']<'2016-12-19')])
]
# list of the values I want to assign for each condition
values = ['week1','week2','week3','week4','week5']
# Let's create a new column and use np.select to assign values to it using our lists as arguments
df_sup['Week'] = np.select(conditions, values)
但是,我遇到了以下错误。 我该如何解决?
另外,我在下面分享数据。
在您的数据conditions
中不是条件列表,因为使用boolean indexing
,所以获取 DataFrames 列表。
因此,对于条件列表,请删除df_sup[
和 last ]
:
conditions = [
(df_sup['Date']>='2016-11-14') & (df_sup['Date']<'2016-11-21'),
(df_sup['Date']>='2016-11-21') & (df_sup['Date']<'2016-11-28'),
(df_sup['Date']>='2016-11-28') & (df_sup['Date']<'2016-12-05'),
(df_sup['Date']>='2016-12-05') & (df_sup['Date']<'2016-12-12'),
(df_sup['Date']>='2016-12-12') & (df_sup['Date']<'2016-12-19')
]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.