如何从包含特定字符串的dataframe中获取列号？

import pandas as pd

# 创建一个示例DataFrame
data = {
    'A': ['foo', 'bar', 'baz'],
    'B': ['qux', 'quux', 'corge'],
    'C': ['grault', 'garply', 'waldo']
}
df = pd.DataFrame(data)

# 要查找的特定字符串
search_string = 'qux'

# 获取包含特定字符串的列号
column_index = None
for column in df.columns:
    if search_string in df[column].values:
        column_index = column
        break

if column_index is not None:
    print(f"包含 '{search_string}' 的列号是: {column_index}")
else:
    print(f"未找到包含 '{search_string}' 的列")