![](/img/trans.png)
[英]Mapping keys from dictionary to values in cases where column being mapped contains a larger string
[英]Mapping values from a dictionary where a column meets a particular criteria
我一直在使用传统的字典映射方法新建一个 Pandas 列,如下所示:
product_costs = {x: 1, y: 2, z: 3}
df['costs'] = df['product'].map(product_costs)
这一直有效,但是最近产品“x”的成本发生了变化——例如,从 4 月 1 日起成本从 1 增加到 4。
我的“df”也有一个日期列,我试图弄清楚如何 map 值 1 ,其中日期列是 4 月之前,而值 4 则日期列是 4 月之后。
我可能可以使用 for 循环迭代地执行此操作,即:
df['costs'] = ''
index = 0
for i in df['product']:
if i == 'x' and df.loc[index, 'date'] < 2021-04-01:
df.loc[index, 'costs'] = 1
index += 1
elif i == 'x' and df.loc[index, 'date'] >= 2021-04-01:
df.loc[index, 'costs'] = 4
index += 1
elif i == 'y':
etc. etc.
...但是,当我确信可以以更简单的方式实现相同的结果时,这似乎相当冗长乏味。 任何人都可以就如何将“日期”元素包含到我的映射中提出解决方案吗?
编辑 - 下面的示例数据
date (dd-mm) product
01-02 x
01-02 y
01-02 z
01-03 x
01-03 y
01-03 z
01-04 x
01-04 y
01-04 z
成为...
date (dd-mm) product cost
01-02 x 1
01-02 y 2
01-02 z 3
01-03 x 1
01-03 y 2
01-03 z 3
01-04 x 4
01-04 y 2
01-04 z 3
np.where()
您可以根据日期条件使用np.where()
。
首先将日期转换为to_datetime()
。 假设您的日期缺少年份( %d-%m
),但您希望年份为 2021:
df['date'] = pd.to_datetime(df['date'], format='%d-%m').apply(lambda x: x.replace(year=2021))
然后 map 与np.where()
以日期为条件:
costs_pre = {'x': 1, 'y': 2, 'z': 3}
costs_post = {'x': 4, 'y': 2, 'z': 3}
df['costs'] = np.where(
df['date'] < '2021-04-01',
df['product'].map(costs_pre),
df['product'].map(costs_post))
# date product costs
# 0 2021-02-01 x 1
# 1 2021-02-01 y 2
# 2 2021-02-01 z 3
# 3 2021-03-01 x 1
# 4 2021-03-01 y 2
# 5 2021-03-01 z 3
# 6 2021-04-01 x 4
# 7 2021-04-01 y 2
# 8 2021-04-01 z 3
np.select()
如果你有多个条件,你可以嵌套np.where()
,但np.select()
会更干净。
例如,如果您的成本在 01-03 和 01-04 再次发生变化:
costs1 = {'x': 1, 'y': 2, 'z': 3}
costs2 = {'x': 4, 'y': 2, 'z': 3}
costs3 = {'x': 100, 'y': 2, 'z': 3}
conditions = [df['date'] < '2021-03-01', df['date'] < '2021-04-01']
choices = [df['product'].map(costs1), df['product'].map(costs2)]
df['costs'] = np.select(conditions, choices, default=df['product'].map(costs3))
# date product costs
# 0 2021-02-01 x 1
# 1 2021-02-01 y 2
# 2 2021-02-01 z 3
# 3 2021-03-01 x 4
# 4 2021-03-01 y 2
# 5 2021-03-01 z 3
# 6 2021-04-01 x 100
# 7 2021-04-01 y 2
# 8 2021-04-01 z 3
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.