如何使用另一個數據幀的熊貓查詢結果過濾熊貓數據幀

Question

我有一個pandas df：

import pandas as pd
df = pd.DataFrame({'col_a' : ['a','a', 'b'], 'col_b': [1,2,3]})
df.index = [4,5,6]

在這個df我應用一個查詢：

df_subset = df.query('col_a == "b"')

現在我有第二個數據框，如下所示：

import numpy as np
df_numpy = pd.DataFrame(np.array([0.1,0.2,0.3]))

這就像原始df但沒有“標識”列（ col_a ）並且值以某種方式轉換（在這個玩具示例中，除以 10）

我想從df_numpy中選擇應用查詢后從df中選擇的相同行。 在這個玩具示例中，第三行。

編輯棘手的部分是df_numpy和df之間的索引值不相同。

有沒有辦法做到這一點 ？

Answer 1

如果有相同的索引值，請使用：

print (df_numpy[df_numpy.index.isin(df_subset.index)])    
     0
2  0.3

編輯：一個想法是在兩者中創建相同的索引值，因為相同的長度：

df = pd.DataFrame({'col_a' : ['a','a', 'b'], 'col_b': [1,2,3]})
df.index = [4,5,6]

df_subset = df.reset_index(drop=True).query('col_a == "b"')

df_numpy = pd.DataFrame(np.array([0.1,0.2,0.3]))


print (df_numpy[df_numpy.reset_index(drop=True).index.isin(df_subset.index)])

     0
2  0.3