• set(集合)数据结构
    • 交集
    • 差集

    set(集合)数据结构

    set(集合)是一个非常有用的数据结构。它与列表(list)的行为类似,区别在于set不能包含重复的值。
    这在很多情况下非常有用。例如你可能想检查列表中是否包含重复的元素,你有两个选择,第一个需要使用for循环,就像这样:

    1. some_list = ['a', 'b', 'c', 'b', 'd', 'm', 'n', 'n']
    2. duplicates = []
    3. for value in some_list:
    4. if some_list.count(value) > 1:
    5. if value not in duplicates:
    6. duplicates.append(value)
    7. print(duplicates)
    8. ### 输出: ['b', 'n']

    但还有一种更简单更优雅的解决方案,那就是使用集合(sets),你直接这样做:

    1. some_list = ['a', 'b', 'c', 'b', 'd', 'm', 'n', 'n']
    2. duplicates = set([x for x in some_list if some_list.count(x) > 1])
    3. print(duplicates)
    4. ### 输出: set(['b', 'n'])

    集合还有一些其它方法,下面我们介绍其中一部分。

    交集

    你可以对比两个集合的交集(两个集合中都有的数据),如下:

    1. valid = set(['yellow', 'red', 'blue', 'green', 'black'])
    2. input_set = set(['red', 'brown'])
    3. print(input_set.intersection(valid))
    4. ### 输出: set(['red'])

    差集

    你可以用差集(difference)找出无效的数据,相当于用一个集合减去另一个集合的数据,例如:

    1. valid = set(['yellow', 'red', 'blue', 'green', 'black'])
    2. input_set = set(['red', 'brown'])
    3. print(input_set.difference(valid))
    4. ### 输出: set(['brown'])

    你也可以用{}符号来创建集合,如:

    1. a_set = {'red', 'blue', 'green'}
    2. print(type(a_set))
    3. ### 输出: <type 'set'>

    集合还有一些其它方法,我会建议访问官方文档并做个快速阅读。