DataFrame获取具有索引和切片的列[重复]

Question

我有一个 DataFrame，我想获得一个包含一些列的新 DataFrame。
我正在使用pandas.DataFrame.iloc()，例如，我想获取第 1 列和第 3 列到第 98 列。
我知道我只能在iloc():new_df = df.iloc[:, [1, 3, 4, ..., 99]] 或切片：new_df = df.iloc[:, 3:99] 中使用索引，但是如何组合它们呢？

当然，有很多方法可以做到：
例如new_df = df.iloc[:, 3:99].drop(df.columns[1], axis=1),
但是有没有办法只使用iloc() 函数？我想获得最短的解决方案。

Answer 1

我找到了这样的解决方案，但我仍然对是否有其他更短的方法感兴趣：
new_df = df.iloc[:, [1] + list(range(3,99))]。

我也可以通过numpy.r_ 实现这一目标：
new_df = df.iloc[:, np.r_[1, 3:99]]。