python - Pandas 通过切片和选择进行索引

Question

无法想象这以前没有被问过，但找不到类似的帖子。问：假设您有一个带有 m 列 (df.shape == (nxm)) 的 pandas 数据框。我想通过切片和选择列来提取列的子集，看起来像： df1 = df.iloc[:,[a,b,c:d]]. 这可能吗？在此示例中，结果应在 df1 中实例化一个数据帧，其中包含数据帧 df 的 a、b 和 c 列直到 d。

标签： pythonpandasdataframeindexingslice

Answer 1

您可以将整数列表或切片传递给iloc，但不能同时传递两者。Numpy 具有r_可用于转换[a,b,c:d]为列表的功能。

import numpy as np

df1 = df.iloc[:, np.r_[a,b,c:d]]