在Pandas DataFrame中选择和使用重复的行

时间:2019-02-05 14:23:09

标签: python pandas dataframe duplicates reformatting

我将以下数据作为熊猫数据框df的一部分:

  qsoName,filterID,aperMag 
0 PSOJ000,3,+19.284586
1 PSOJ007,2,+20.334393
2 PSOJ007,3,+20.226970
3 PSOJ007,4,+20.288778   
4 PSOJ007,5,+20.189209
5 PSOJ011,2,+21.037594
6 PSOJ011,4,+20.642813            
7 PSOJ011,5,+20.760576 

我想为df['aperMag']的一个值选择df['qsoName']的不同值,默认值为-999.99999,例如

 PSOJ000,-999.99999,+19.284586,-999.99999,-999.99999
 PSOJ007,+20.334393,+20.226970,+20.288778,+20.189209
 PSOJ011,+21.037594,-999.99999,+20.642813,+20.760576 

这感觉df["qsoName"].duplicated()应该起作用,但是DataFrame的重新格式化是另一个关键部分。

1 个答案:

答案 0 :(得分:3)

DataFrame.pivotDataFrame.fillna一起使用:

df = df.pivot('qsoName','filterID','aperMag').fillna(-999.99999)

或者用set_index和参数fill_value的{​​{3}}:

df = df.set_index(['qsoName','filterID'])['aperMag'].unstack(fill_value=-999.99999)

print (df)
filterID           2           3           4           5
qsoName                                                 
PSOJ000  -999.999990   19.284586 -999.999990 -999.999990
PSOJ007    20.334393   20.226970   20.288778   20.189209
PSOJ011    21.037594 -999.999990   20.642813   20.760576

最后必要时:

df = df.reset_index().rename_axis(None,axis=1)
print (df)
   qsoName           2           3           4           5
0  PSOJ000 -999.999990   19.284586 -999.999990 -999.999990
1  PSOJ007   20.334393   20.226970   20.288778   20.189209
2  PSOJ011   21.037594 -999.999990   20.642813   20.760576

编辑:

问题是某些数据对(qsoNamefilterID)重复,因此需要unstack

df = df.pivot_table(index='qsoName',
                    columns='filterID', 
                    values='aperMag', 
                    fill_value=-999.99999, 
                    aggfunc='mean')