[英]Get first two rows from first two groups
考慮數據幀df
mux = pd.MultiIndex.from_arrays([
list('aaaabbbbbccdddddd'),
list('tuvwlmnopxyfghijk')
], names=['one', 'two'])
df = pd.DataFrame({'col': np.arange(len(mux))}, mux)
df
col
one two
a t 0
u 1
v 2
w 3
b l 4
m 5
n 6
o 7
p 8
c x 9
y 10
d f 11
g 12
h 13
i 14
j 15
k 16
如果我按索引的第一級分組,如何優雅地獲取前兩組的前兩行:
col
one two
a t 0
u 1
b l 4
m 5
選項1
您可以使用list comp和pd.concat
:
pd.concat([g.head(2) for _, g in df.groupby(level=0)][:2])
col
one two
a t 0
u 1
b l 4
m 5
由於列表comp完成是一個不必要的開銷,你可以使用itertools.takewhile
來防止這種情況。
it = itertools.takewhile(lambda x: x[0] < 2, enumerate(df.groupby(level=0)))
pd.concat([g.head(2) for _, (_, g) in it])
col
one two
a t 0
u 1
b l 4
m 5
選項2
我能想到的另一個可能的解決方案是預過濾你的df以保留僅為索引級別0的前兩個值的行,然后執行groupby。
# https://stackoverflow.com/a/46900625/4909087
df.loc[df.index.levels[0][:2].values].groupby(level=0).head(2)
col
one two
a t 0
u 1
b l 4
m 5
看起來很討厭,但這是我嘗試過的
df.groupby(level=['one']).head(2)[:4]
col
one two
a t 0
u 1
b l 4
m 5
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.