如何在熊猫中跳过两个不同的范围

Question

我想隔离一个位于cvs文件中间的数据集，但我不知道如何隔离。

我试过了

data_df_2 = pd.read_csv('data_set.csv', skiprows=range(1:8,21:114)

和

data_df_2 = pd.read_csv('data_set.csv', skiprows=range(1,8), skiprows=range(21,114)

它们都不起作用。解决我的问题的正确方法是什么？

Answer 1

pandas.read_csv doc声明skiprows需要是类似列表的、整型的或可调用的。因此，您可以创建列表理解来生成要排除的范围。

pd.read_csv('data_set.csv', skiprows=([i for i in range(1, 8)] + [j for j in range(21, 114)]))

Answer 2

为了便于显示，我更改了您的值，但概念是相同的。

skip_1 = list(range(1, 8, 1))
skip_2 = list(range(12, 15, 1))
skip_idxs_list = skip_1 + skip_2
df_slice = df.iloc[~df.index.isin(skip_idxs_list)]
df_slice.head(20)

​

​