Python:在满足条件的情况下返回列索引

def find_column_index(matrix, condition):
    for row in matrix:
        for i, value in enumerate(row):
            if value == condition:
                return i
    return -1  # 如果没有找到满足条件的列索引，返回-1

# 示例数据
matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

condition = 5
column_index = find_column_index(matrix, condition)
if column_index != -1:
    print(f"满足条件的列索引为: {column_index}")
else:
    print("未找到满足条件的列索引")