獲取以給定字符串開頭的列中的唯一項

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']

對於熊貓解決方案，請使用：

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