SyntaxError: invalid syntax

def extract(data=None,columns=None):
    if data is not None and columns is not None:
        
        #return df_data[columns.capitalize()]
        return df_data[ i.capitalize() for i in columns]
 
    else:
        print('Error: input arguments are not provided properly extract(data, columns)')
 
 
  File "<ipython-input-171-5be15b70c54c>", line 6
    return df_data[ i.capitalize() for i in columns]
                                                 ^
      SyntaxError: invalid syntax

      
User ID  Gender Age EstimatedSalary Purchased
0    15624510    Male  19           19000         0
1    15810944    Male  35           20000         0
2    15668575  Female  26           43000         0
3    15603246  Female  27           57000         0
4    15804002    Male  19           76000         0
..        ...     ...  ..             ...       ...
395  15691863  Female  46           41000         1
396  15706071    Male  51           23000         1
397  15654296  Female  50           20000         1
398  15755018    Male  36           33000         0
399  15594041  Female  49           36000         1
 
[400 rows x 5 columns]
 
 
df_target0=extract()
df_target1=extract(data=df_data)
df_target2=extract(columns=["User ID"])
df_target3=extract(data=df_data, columns=["User ID"])
df_target4=extract(data=df_data, columns=["User ID","Purchased"])
df_target5=extract(data=df_data, columns=["gender","Purchased"])
df_target6=extract(columns="User ID",data=df_data,)
 
Ожидаемый результат:
Error: input arguments are not provided properly extract(data, columns)
Error: input arguments are not provided properly extract(data, columns)
Error: input arguments are not provided properly extract(data, columns)
Error: input arguments are not provided properly extract(data, columns)
 
1) User ID
0    15624510
1    15810944
2    15668575
2)      User ID Purchased
0    15624510         0
1    15810944         0
2    15668575         0
3)    Purchased
0           0
1           0
2           0

[i.capitalize() for i in columns] # это генератор
list([i.capitalize() for i in columns]) # это список

return df_data[list([ i.capitalize() for i in columns])]

return df_data[[ i.capitalize() for i in columns]]

...
    if isinstance(data, pd.DataFrame) and isinstance(columns, list):