如何在熊猫数据框中选择多范围的行

Question

给出一个索引为从0到的熊猫数据框的例子30。我想选择索引[0:5]、[10:15]和 的几个范围内的行[20:25]。

怎么做？

Answer 1

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

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], :]