使用熊猫创建新列的条件拆分

时间:2019-04-26 06:59:51

标签: python python-3.x pandas

我有一个数据,其中在“类型”列中有一组货币。我根据这些货币创建了新列。但是我不确定如何给条件,例如类型是否为btc,然后创建这些列,就像明智的那样,我在该列中有更多类别。示例数据如下所示。

给出:

Type    Last                 
ada     3071.56         
ada     3097.82          
btc     1000.00
ada     2000.00
btc     3000.00
eur     1000.00
eur     1500.00

预期:

Type    Last        changbtwread_ada    changbtwread_btc  changbtwread_eur         
ada     3071.56          Nan                 Nan               Nan
ada     3097.82          26.0                Nan               Nan
btc     1000.00          Nan                 Nan               Nan 
ada     2000.00         -1097.82             Nan               Nan
btc     3000.00          Nan                 2000              Nan
eur     1000.00          Nan                 Nan               Nan
eur     1500.00          Nan                 Nan               500

我试图在一种类型的读取之间创建此更改的代码如下:

df['changbtwread'] = df['Last'].diff()

但是我想要的是所有类型的东西。

1 个答案:

答案 0 :(得分:0)

您可以按唯一值循环,并与由f-string s创建的新列产生差异:

for v in df['Type'].unique():
    df[f'changbtwread_{v}'] = df.loc[df['Type'].eq(v), 'Last'].diff()
print (df)
  Type     Last  changbtwread_ada  changbtwread_btc  changbtwread_eur
0  ada  3071.56               NaN               NaN               NaN
1  ada  3097.82             26.26               NaN               NaN
2  btc  1000.00               NaN               NaN               NaN
3  ada  2000.00          -1097.82               NaN               NaN
4  btc  3000.00               NaN            2000.0               NaN
5  eur  1000.00               NaN               NaN               NaN
6  eur  1500.00               NaN               NaN             500.0