python - 使用 iloc() 根据范围和单个整数选择列

Question

鉴于下面的数据框，我想选择 A 列、D 列到 F 列。

import numpy as np
import pandas as pd
df = pd.DataFrame(np.random.randint(0,100,size=(5, 6)), columns=list('ABCDEF'))

在 R 中我可以说

df[ , c(1, 4:6)]

我想要 Pandas 类似的东西。与

df.iloc[:, slice(3, 6, 1)]

我获得了 D 至 F 列。但是如何添加 A 列？

Answer 1

您使用np.r_来传递切片和索引的组合。由于您似乎知道标签，因此可以使用 get_loc 来获取 iloc 索引。

import numpy as np

idx = np.r_[df.columns.get_loc('A'), 
            df.columns.get_loc('D'):df.columns.get_loc('F')+1]
# idx = np.r_[0, 3:6]

df.iloc[:, idx]
    A   D   E   F
0  38  71  62  63
1  60  93  72  94
2  57  33  30  51
3  88  54  21  39
4   0  53  41  20