In [1]:

    

import pandas as pd


    

In [2]:

    

a = pd.read_csv("/Users/weilu/Research/server/may_week1_2020/Final_2Sm.csv")


    

In [4]:

    

a = a.query("Class == 'α'")


    

In [5]:

    

pdb_list = []
for pdb in a.PDB.unique():
    pdb = str(pdb)
    if "(" in pdb:
        pass
    elif pdb == "nan":
        pass
    else:
        pdb_list.append(pdb)


    

In [7]:

    

second_test_test = a.query("PDB in @pdb_list").reset_index(drop=True).iloc[:, :23]


    

In [ ]: