对同一数据帧的不同部分进行操作

时间:2018-05-17 11:37:27

标签: python pandas dataframe

假设有一个数据帧:

    kind  value
0      1      1
1      1      2
2      1      3
3      1      4
4      1      5
5      2      6
6      2      7
7      2      8
8      2      9
9      2      10

我们可以对数据帧的过滤部分执行某些操作:

df.loc[df['kind']==1, 'value'] = df.loc[df['kind']==1, 'value'] * 2

假设它们的大小相等,如何执行涉及同一数据帧的两个或多个部分的计算?像这样:

df.loc[df['kind']==1, 'value'] = 
    df.loc[df['kind']==1, 'value'] * df.loc[df['kind']==2, 'value']

(这不起作用)

2 个答案:

答案 0 :(得分:5)

试试这个:

In [107]: df.loc[df['kind']==1, 'value'] *= df.loc[df['kind']==2, 'value'].values

In [108]: df
Out[108]:
   kind  value
0     1      6
1     1     14
2     1     24
3     1     36
4     1     50
5     2      6
6     2      7
7     2      8
8     2      9
9     2     10

答案 1 :(得分:2)

使用:

m = df['kind']==1
df.loc[m, 'value'] = df.loc[m, 'value'].values * df.loc[df['kind']==2, 'value'].values
print (df)
   kind  value
0     1      6
1     1     14
2     1     24
3     1     36
4     1     50
5     2      6
6     2      7
7     2      8
8     2      9
9     2     10