当前位置:首页 > 行业动态 > 正文

pandas 判断列是否存在

在 Pandas 中,可以使用 in 运算符来判断列是否存在于 DataFrame 中,下面是一个详细的示例:

判断列是否存在

要判断一个列是否存在于 DataFrame 中,可以使用 in 运算符结合 columns 属性来实现,下面是具体的步骤:

pandas 判断列是否存在  第1张

1、导入 Pandas 库:

import pandas as pd

2、创建一个 DataFrame:

data = {'Name': ['John', 'Emma', 'Mike'],
        'Age': [25, 28, 30],
        'City': ['New York', 'London', 'Paris']}
df = pd.DataFrame(data)

3、使用 in 运算符和 columns 属性判断列是否存在:

pandas 判断列是否存在  第2张

column_name = 'Age'
if column_name in df.columns:
    print(f"The column '{column_name}' exists in the DataFrame.")
else:
    print(f"The column '{column_name}' does not exist in the DataFrame.")

运行以上代码,将输出以下结果:

The column 'Age' exists in the DataFrame.

如果将 column_name 的值改为不存在的列名,'Gender',则输出结果为:

The column 'Gender' does not exist in the DataFrame.

通过这种方式,你可以方便地判断列是否存在于 DataFrame 中。

pandas 判断列是否存在  第3张

0