python - 根据其他数据框中的值检索一行数据框中的值

Question

我有多个 DataFrame，每个 DataFrame 都包含一个名为“location”的行和另一个名为“value”的行(均构成索引)。例如，假设我有以下 2 个:

df1 = pd.DataFrame(np.array([[-4,2,5],['nyc','sf','chi']]), columns=['col1','col2','col3'], index=['value','location'])

df2 = pd.DataFrame(np.array([[5,0,-3],['nyc','sf','chi']]), columns=['col1','col2','col3'], index=['value','location'])

DataFrame 将存放在我可以遍历的字典中。最终，我想在单独的 DataFrame 中检索每个“位置”的“值”列表。所以所需的输出看起来像:

这是一个玩具示例，而我的真实示例将有更多的 DataFrames 并且源 DataFrames 除了我感兴趣的 2 个关键行之外还有其他行

Answer 1

我会推荐 set_index 和 concat:

(pd.concat([df.T.set_index('location')['value'] for df in [df1, df2]], axis=1)
   .T
   .reset_index(drop=True))

location nyc sf chi
0         -4  2   5
1          5  0  -3