创建一个新的dataframe，当某些行与另一个datafram匹配时，该dataframe包含一个dataframe的两列

Question

我有两个大数据帧。”“基因”的结构是这样的（虽然不止三列）：

Mutation ID   CDS         Mutation AA
COSM1000525   123         V617A 
COSM1003371   234         S517A
COSM1003372   456         T417A
COSM1004880   567         K317A
COSM1004881   1234        R610A
COSM1004883   1234        V617A
COSM1004884   543         W617A
COSM1006396   132         Y617A
COSM1007340   7654        V617A

“output”是“gene”的一部分，只有一列Mutation ID：

Mutation ID             
COSM1000525   
COSM1003372                   
COSM1004881   

我想创建一个新的数据帧，当突变ID与“output”中的那些匹配时，它有两列（“gene”的Mutation ID和Mutation AA）

如何在python中实现这一点

Answer 1

熊猫Indexing and Selecting Data是一个很好的向导

mut_filter = set(output['Mutation ID'])
df = gene.loc[gene['Mutation'].isin(mut_filter), ['Mutation ID', 'Mutation AA']]