按特定列中的索引对熊猫数据框进行切片

Question

假设我有一个 Pandas datafra df并且我有A,B,C列。

我想从B列中获取索引' idx=[1,10,12,17] - 这是怎么做的？

我试过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