我有以下代码设置,调用和groupBy并应用于Python Pandas数据帧。
奇怪的是,如果没有完全扭曲输出(如调试中所示),我就无法按行(如df.loc[2:5]
)对分组数据进行切片,如何才能删除行并获取该行以提供所需的输出?
任何帮助都是非常感谢的,我正在一个更大的例子上运行这个更复杂的函数,但已经指出了问题的行切片!
代码:
import pandas as pd
df = pd.DataFrame({'one' : ['AAL', 'AAL', 'AAPL', 'AAPL'], 'two' : [1, 2, 3, 4]})
def net_func(df):
df_res = daily_func(df, True)
df_res_valid = daily_func(df, False)
df_merge = pd.merge(df_res, df_res_valid)
return df_merge
def daily_func(df, bool_param):
# df.drop(df.head(1).index, inplace=True)
# df = df[1:1]
# df.iloc[1:1,:]
# df.loc[1:1,:]
if bool_param:
df['daily'+str(bool_param)] = 1
else:
df['daily'+str(bool_param)] = 0
return df
print df.groupby('one').apply(net_func)
电流输出:
one two dailyTrue dailyFalse
one
AAL 0 AAL 1 1 0
1 AAL 2 1 0
AAPL 0 AAPL 1 1 0
1 AAPL 2 1 0
期望输出:
one two dailyTrue dailyFalse
one
AAL 1 AAL 2 1 0
AAPL 1 AAPL 2 1 0
理想情况下,我希望能够为每个组按行切片,例如df.loc[3:5]
-这将是完美的!
我试过如下评论:
使用df.drop(df.head(1).index, inplace=True)
输出:
Empty DataFrame
Columns: [one, two, dailyTrue, dailyFalse]
Index: []
更新:还尝试使用df = df[1:1]
输出:
Empty DataFrame
Columns: [one, two, dailyTrue, dailyFalse]
Index: []
更新还尝试了df.iloc[1:1,:]
:
one two dailyTrue dailyFalse
one
AAL 0 AAL 1 1 0
1 AAL 2 1 0
AAPL 0 AAPL 1 1 0
1 AAPL 2 1 0
以及df.loc[1:1,:]
:
one two dailyTrue dailyFalse
one
AAL 0 AAL 1 1 0
1 AAL 2 1 0
AAPL 0 AAPL 1 1 0
1 AAPL 2 1 0
考虑使用横截面切片,^{} 在
groupby().apply()
之后,相应地指定每个键:或者,对元组列表使用multiple indexing:
还有slice(在熊猫0.14中引入):
相关问题 更多 >
编程相关推荐