Pandas：查找excel表中是否存在列名？答案

【问题标题】：Pandas: find if a column name exists in excel sheet?Pandas：查找excel表中是否存在列名？
【发布时间】：2021-08-27 00:18:16
【问题描述】：

使用熊猫：我正在阅读 1 个 excel 表并从中收集数据，然后创建一个新的 excel 文档。

我的代码只有在用户选择一个具有准确、特定字符串名称的 Excel 工作表时才有效。我需要能够检查列名是否存在于 Excel 工作表中。我现在拥有的：

db = pandas.read_excel(full_path_with_file_name, sheetname=sheet_name)
list_values = db['column_name'].values

我获取该值列表并将其与 xlsxwriter 一起使用。

在我从该列中收集值之前，我想先检查该列是否存在于 excel 中。如果该列不存在，我只想简单地将“未知”分配为新 xlsx 的值。

我尝试过各种形式的：

if db['column_names']:
     list_values = db['column_name'].values
else:
     list_values = # just 'unknown' or 'not found'

或者类似的版本。

如何正确检查excel列名是否存在？

【问题讨论】：

嘿！如果 df.columns 中的“列名”可以使用：

标签： python excel pandas

【解决方案1】：

如果不打开电子表格，就无法对其进行测试。试试这个：

xls = pd.read_excel('test.xls', sheet_name="Sheet1")
df = pd.DataFrame({'COL0': np.nan*5}, np.arange(5))

columnsToSeek = ['COL2', 'COL3']

for col in columnsToSeek:
    if col in xls.columns:
        df[col] = xls[col]
    else:
        df[col] = 'UNKNOWN'
    
df

    COL0    COL2    COL3
0   NaN     cc      UNKNOWN
1   NaN     dd      UNKNOWN
2   NaN     NaN     UNKNOWN
3   NaN     NaN     UNKNOWN
4   NaN     NaN     UNKNOWN

【讨论】：