python - 带有偏移的 Pandas 头

Question

例如，获取偏移量 Y 之后的前 X 行的建议方法是什么？我目前正在做的是:

offset, limit = 2, 2
df=pd.DataFrame([{'a':1}, {'a': 2}, {'a':3}, {'a': 4}, {'a':5}])
df.head(offset+limit)[offset:]
#    a
# 2  3
# 3  4

有更好的方法吗？

Answer 1

您可以使用 df.iloc在这里。

df.iloc[offset: offset+limit]

   a
2  3
3  4