python - 从 Pandas 系列中创建一个集合

Question

我有一个从 Kaggle 的 San Fransico Salaries 中提取的数据框:https://www.kaggle.com/kaggle/sf-salaries我希望创建一组列的值，例如“状态”。

这是我尝试过的，但它带来了所有记录的列表而不是集合(sf 是我命名数据框的方式)。

a=set(sf['Status'])
print a

根据这个网页，这应该可以工作。 How to construct a set out of list items in python?

Answer 1

如果您只需要获取唯一值列表，您可以使用 unique方法。如果你想拥有 Python 的集合，那么 set(some_series)

In [1]: s = pd.Series([1, 2, 3, 1, 1, 4])

In [2]: s.unique()
Out[2]: array([1, 2, 3, 4])

In [3]: set(s)
Out[3]: {1, 2, 3, 4}

但是，如果您有 DataFrame，只需从中选择系列 (some_data_frame['<col_name>'])。