如何使用行在Pandas中选择范围

Question

我有一个Pandas数据帧.

我在另一个进程中从该数据帧中选择了一行.

在另一种方法中,我现在需要从行所在的数据帧中选择一个范围,如果有这么多,则返回55行.

这是一些伪代码,希望它有所帮助:

df = DataFrame from csv

row = df[3454]

index = row.index
start = max(0, index - 55)
end = max(1, index)
dfRange = df[start:end]

Answer 1

这应该做到这一点

integer_location = np.where(df.index == 3454)[0][0]
start = max(0, integer_location - 55)
end = max(1, integer_location)
dfRange = df.iloc[start:end]

此链接有更多信息 获取满足条件的Pandas DataFrame行的整数索引？