pandas 列筛选

def pandas_test():

    """

    (1) str.contains -> 字符串包含

    (2) isin -> 列表包含

    :return:

    """

    temp = [{'name': 'cat', 'age': 12},

            {'name': 'pig', 'age': 13},

            {'name': 'mouse', 'age': 10},

            {'name': 'monkey', 'age': 20}]

    df = pd.DataFrame(temp)

    df1 = df[df['name'].str.contains('mo')]  # 筛选出 name 这一列中包含 mo 的所有行

    df2 = df[~df['name'].str.contains('mo')]  # 筛选出 name 这一列中不包含 mo 的所有行

    df3 = df[df.name.isin(['mouse', 'chick', 'duck'])]  # 筛选出 name 这一列存在于指定列表的所有行

    df4 = df[~df.name.isin(['mouse', 'chick', 'duck'])]  # 筛选出 name 这一列不存在于指定列表的所有行

    print(df1)

    print(df2)

    print(df3)

    print(df4)

if __name__ == '__main__':

    pandas_test()