Pandas：根据列是否存在的条件选择DataFrame中的列

在本文中，我们将介绍如何在Pandas中选择具有特定条件的列。

阅读更多：Pandas 教程

需求描述

我们需要从DataFrame中选择存在特定列的列。

例如，我们有以下DataFrame：

我们需要选择所有具有“电话号码”列的列。

实现方法

首先，我们需要定义一个函数，它可以返回所有具有特定列的列的名称。

def select_columns_with_columns_names(df, column_list):
    """
    返回具有给定列名的DataFrame列
    df: DataFrame
    column_list: 包含列名的列表
    """
    return df[[col for col in column_list if col in df.columns]]

现在，我们可以使用此函数来选择所有具有“电话号码”列的列，如下所示：

import pandas as pd

df = pd.DataFrame({
                   '编号': [1, 2, 3],
                   '名称': ['王小明', '张小玲', '刘小芳'],
                   '城市': ['北京', '上海', '广州'],
                   '电话号码': [18888888888, 17777777777, 16666666666]
                  })

columns = ['名称', '城市', '电话号码']
df_filtered = select_columns_with_columns_names(df, columns)

现在，我们可以查看选择的DataFrame：

print(df_filtered.head())

输出为：

现在，我们选择了“名称”、“城市”和“电话号码”列，因为它们都存在于DataFrame中。

总结

在Pandas中，我们可以使用自定义函数来选择具有特定条件的列。在本文中，我们已经介绍了如何根据列是否存在的条件选择DataFrame中的列。我们使用了定义的函数并提供了演示代码，以帮助你更好地理解如何在实践中实现此需求。