python - Pandas Dataframe 选择多个不连续的列/切片

Question

我有超过 100 列的数据框。我正在尝试选择列 0~32 和 #83。看起来 1 slice 可以很好地处理下面的代码。

df_new = df[df.columns[0:32]]

但它不适用于下面的 2 个切片代码。我该如何解决这个问题？

df_new = df[df.columns[0:32, 83]]

Answer 1

将 np.r_ 索引器与 iloc 结合使用，如下所示:

df.iloc[:, np.r_[0:32, 83]]

---

np.r_[0:32, 83]

array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16,
       17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 83])