我将以下数据作为熊猫数据框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的重新格式化是另一个关键部分。
答案 0 :(得分:3)
将DataFrame.pivot
与DataFrame.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
编辑:
问题是某些数据对(qsoName
,filterID
)重复,因此需要unstack
:
df = df.pivot_table(index='qsoName',
columns='filterID',
values='aperMag',
fill_value=-999.99999,
aggfunc='mean')