pd.read_csv 如何判断列是否存在

您好，关于您的问题，可以使用 Pandas 库中的 `pd.read_csv()` 函数中的 `usecols` 参数，它可以用来指定需要读取的列，如果指定的列不存在，则会引发 `KeyError` 错误。可以根据这个错误来判断列是否存在。下面是一个示例代码：

import pandas as pd

filename = 'data.csv'
required_cols = ['col1', 'col2', 'col3']

try:
    df = pd.read_csv(filename, usecols=required_cols)
except KeyError as e:
    print(f"列 '{str(e)}' 不存在！")

在这个示例中，`filename` 变量是 CSV 文件的路径，`required_cols` 变量是需要读取的列的列表。如果文件中不存在指定的列，则会打印错误消息。