在特定列中按索引切片熊猫数据框

Question

假设我有一个df的熊猫数据，并且我有一个A,B,C的列。

我要从idx=[1,10,12,17]列中获取索引'B-如何完成？

我尝试过df[idx,"B"], df.iloc[idx,"B"], df.loc[idx,"B"]

Answer 1

您可以.loc或.iloc。

idx = [1,10,12,17]
df = pd.DataFrame(np.random.rand(20, 3), columns=['A', 'B', 'C'])

df.loc[idx, 'B']
df.iloc[idx, 1]

结果：

1     0.532895
10    0.197801
12    0.978466
17    0.847575
Name: B, dtype: float64