有条件的熊猫唯一值

Question

我正在使用 pandas DataFrame，我需要循环遍历列的唯一值。 这些列可能包含我不想循环的值，例如 ''

通常我会这样做：

edges = [edge for edge in estados['EDGE'].unique() if edge != '']
for edge in edges:
    pass

我的问题是，是否有一种比理解列表更流行的方式来构建列表。

喜欢：

estados['EDGE'].unique().exclude('')

谢谢

注意： 我在以下位置寻找解决方案： nunique excluding some values in pandas Python pandas unique value ignoring NaN 但这些解决方案甚至不如我的简洁。

Answer 1

您可以使用 NOT 运算符~:

estados[~estados['EDGE'] == '']['EDGE'].dropna().unique()

或使用.ne：

estados[estados['EDGE'].ne('')]['EDGE'].dropna().unique()

Answer 2

您可以将pd.Series.mask 与pd.Series.dropna 和pd.Series.unique 一起使用

m = estados['EDGE']==''
estados['EDGE'].mask(m).dropna().unique()

或pd.Series.notna

m = (estados['EDGE']!='') & (estados['EDGE'].notna())

estados['EDGE'][m]