我有一个像这样的pandas DataFrame:
From To Val
GE VD 1000
GE VS 1600
VS VD 1500
VS GE 600
VD GE 1200
VD VS 1300
我想将“from”或“to”列中没有“GE”的每一行替换为两行,一行在“from”列中有“GE”,另一行在“to”中有“GE”. “专栏.
在上面的例子中,我将用以下两行替换第三行:
GE VD 1500
VS GE 1500
我尝试使用“apply”但我无法弄清楚如何返回正确的数据框.例如
def myfun(row):
if "GE" not in (row["from"], row["to"]):
row1=pd.DataFrame(row).T
row2=row1.copy()
row1["from"]="GE"
row2["to"]="GE"
return pd.concat([row1, row2])
else:
return pd.DataFrame(row).T
给出一个奇怪的结果:
>> df.apply(myfun, axis=1)
Val from to
0 Val from to
1 Val from to
2 Val from to
3 Val from to
4 Val from to
5 Val from to
虽然我的功能看似正确:
>> myfun(df.loc[5])
Val from to
5 13 GE VD
5 13 VS GE
通过在两个子数据帧中过滤我的数据帧,我可以想到一种方法,一个行需要重复,另一个需要重复.然后复制第一个数据帧,进行更改并将所有三个DF整理在一起.但它很难看.有谁能建议更优雅的方式?
换句话说,应用函数可以返回一个DataFrame,就像在R中我们会用ddply做的那样吗?
谢谢