<SYSTEM_TASK:>
Given the following text problem statement, write Python code to implement the functionality described below in problem statement
<END_TASK>
<USER_TASK:>
Problem:
I am trying to extract rows from a Pandas dataframe using a list of row names, but it can't be done. Here is an example
<ASSISTANT_TASK:>
Python Code:
import pandas as pd
import io
data = io.StringIO("""
rs  alleles  chrom  pos strand  assembly#  center  protLSID  assayLSID
TP3      A/C      0    3      +        NaN     NaN       NaN        NaN
TP7      A/T      0    7      +        NaN     NaN       NaN        NaN
TP12     T/A      0   12      +        NaN     NaN       NaN        NaN
TP15     C/A      0   15      +        NaN     NaN       NaN        NaN
TP18     C/T      0   18      +        NaN     NaN       NaN        NaN
""")
df = pd.read_csv(data, delim_whitespace=True).set_index('rs')
test = ['TP3', 'TP7', 'TP18']
def g(df, test):
    return df.loc[test]
result = g(df, test)
<END_TASK>