如何使用熊猫对数据框进行子集化？ [复制]

Question

我想对可能相邻也可能不相邻的列进行子集化。在这个试点案例中，只有 39 列，因此写出完整的列名是可能的，但很乏味。我要做的只是基于索引的子集列。

df_sub = df.iloc[:,[10,11,29:37]]

但我收到此错误：

File "<ipython-input-46-d9362578c302>", line 1
    df_sub = df.iloc[:,[10,11,29:37]]
                                ^
SyntaxError: invalid syntax

有没有办法对列索引可能不相邻的列进行子集化？

在谷歌上搜索做这样的事情只向我展示了基于列名、彼此相邻的索引或单独输入每个索引的子集方法，例如here。如果有 100 列的混合物怎么办？

Answer 1

你可以试试这样的：

g = lambda x,y: range(x,y)

[10,11,*g(29,37)]

---

#[10, 11, 29, 30, 31, 32, 33, 34, 35, 36]

---

时间对比：

%%timeit -n 1000000
np.r_[10,11,29:37]

每个循环 12.4 µs ± 123 ns（平均值 ± 标准偏差，7 次运行，每次 1000000 次循环）

%%timeit -n 1000000
g = lambda x,y: range(x,y)
[10,11,*g(29,37)]

每个循环 338 ns ± 3.94 ns（平均值 ± 标准偏差，7 次运行，100000 次循环 每个）