从一行 Dataframe 中获取数据子集

Question

假设我有一个包含“A”、“B”、“C”列的数据框 df 现在我只想提取 df 的第 2 行和“B”和“C”列。 最有效的方法是什么？

你能告诉我为什么 df.ix[2, ['B', 'C']] 不起作用吗？

谢谢！

Answer 1

row_2 = df[['B', 'C']].iloc[1]

或者

# Convert column to 2xN vector, grab row 2 row_2 = list(df[['B', 'C']].apply(tuple, axis=1))[1]

Answer 2

考虑数据帧df

df = pd.DataFrame(np.arange(9).reshape(3, 3), list('xyz'), list('ABC'))
df

   A  B  C
x  0  1  2
y  3  4  5
z  6  7  8

如果你想维护一个数据框

df.loc[df.index[[1]], ['B', 'C']]

   B  C
y  4  5

如果你想要一个系列

df.loc[df.index[1], ['B', 'C']]

B    4
C    5
Name: y, dtype: int64