Identify columns in a DataFrame that could be a unique key

def _find_key_cols(df):
    """Identify columns in a DataFrame that could be a unique key"""
    
    keys = []
    for col in df:
        if len(df[col].unique()) == len(df[col]):
            keys.append(col)
    return keys