按特定列中的索引對熊貓數據框進行切片

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