熊猫切片排除结束

时间:2017-08-05 15:49:50

标签: python pandas indexing

使用loc切片数据框时,

df.loc [开始:结束]

包括开始和结束。使用loc时是否有一种简单的方法可以排除结尾?

3 个答案:

答案 0 :(得分:8)

loc包括开始和结束,一个不太理想的工作是获取索引位置并使用iloc切片数据框(假设您没有重复索引):< / p>

df=pd.DataFrame({'A':[1,2,3,4]}, index = ['a','b','c','d'])

df.iloc[df.index.get_loc('a'):df.index.get_loc('c')]

#   A
#a  1
#b  2

df.loc['a':'c']

#   A
#a  1
#b  2
#c  3

答案 1 :(得分:3)

所有答案均未解决end不属于索引的情况。 更为通用的解决方案是简单地将索引与startend进行比较,这样您就可以强制将其中的任何一个都包括在内。

df[(df.index >= start) & (df.index < end)]

例如:

>>> import pandas as pd
>>> import numpy as np

>>> df = pd.DataFrame(
    {
        "x": np.arange(48),
        "y": np.arange(48) * 2,
    },
    index=pd.date_range("2020-01-01 00:00:00", freq="1H", periods=48)
)

>>> start = "2020-01-01 14:00"
>>> end = "2020-01-01 19:30" # this is not in the index

>>> df[(df.index >= start) & (df.index < end)]

                    x   y
2020-01-01 14:00:00 14  28
2020-01-01 15:00:00 15  30
2020-01-01 16:00:00 16  32
2020-01-01 17:00:00 17  34
2020-01-01 18:00:00 18  36
2020-01-01 19:00:00 19  38

答案 2 :(得分:2)

我能想到的最简单的是df.loc[start:end].iloc[:-1]

切断最后一个。