如何在pandas数据帧中选择多行范围

Question

给出一个带有从0到30索引的pandas数据帧的示例。我想选择几个索引范围内的行[0:5]，[10:15]和[20:25]。

怎么做？

Answer 1

假设您有一个随机的pandas DataFrame，包含30行和4列，如下所示：

import numpy as np
import pandas as pd
df = pd.DataFrame(np.random.randint(0,30,size=(30, 4)), columns=list('ABCD'))

然后，您可以使用np.r_索引行[0:5]，[10:15]和[20:25]的行范围，如下所示：

df.loc[np.r_[0:5, 10:15, 20:25], :]