Получить уникальные элементы в столбце, начиная с заданной строки

Question

Рассмотрим столбец с его уникальными значениями:

df['something'].unique() =array(['aa','bb','a','c']).

Теперь я хочу узнать, какой из элементов начинается с a. Мой ожидаемый ответ

'aa','a'

Answer 1

Я думаю, что здесь самое простое использование понимания списка с фильтрацией:

out = [x for x in df['something'].unique() if x.startswith('a')]
print (out)
['aa', 'a']

Для pandas использования решения:

s = pd.Series(df['something'].unique())
out = s[s.str.startswith('a')].tolist()
print (out)
['aa', 'a']