基于Pandas中列值的条件乘法

时间:2018-10-30 16:16:29

标签: pandas pandas-groupby

我有一个数据框,如下所示:

index  client year  value
  1      A    2011    5
  2      A    2012    10
  ...
  8      A    2018    7
  9      B    2011    14
  10     B    2012    54
  ...
  ...    Z    2011    5

我需要使用与年份相关的值乘以这些值。我的系列如下:

2011  2
2012  2.5
2013  3
2014  3.5
...
2018  5.5

我需要所有年份的值乘以这些年份相关的值。例如。看起来像

index  client year  value
  1      A    2011    10
  2      A    2012    25
 ...etc

我现在编写了一个循环来执行此操作,但这远非优雅而有效。如何高效,优雅地做到这一点?

2 个答案:

答案 0 :(得分:5)

使用reindex

df.value*=s.reindex(df.year).values
df
Out[44]: 
   index client  year  value
0      1      A  2011   10.0
1      2      A  2012   25.0

答案 1 :(得分:3)

如果您的系列如下:

>>> s
0
2011    2.0
2012    2.5
2013    3.0
2014    3.5
2018    5.5
Name: 1, dtype: float64

然后您可以使用map进行此操作:

df['value'] = df['value'].mul(df.year.map(s))

您会得到:

>>> df
   index client  year  value
0      1      A  2011   10.0
1      2      A  2012   25.0
2      8      A  2018   38.5
3      9      B  2011   28.0
4     10      B  2012  135.0
....