如何使用regex在csv中检查列中的有效值？

import csv
import re

def check_valid_values(csv_file, column_index, regex_pattern):
    valid_values = []
    invalid_values = []

    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            if len(row) > column_index:
                value = row[column_index]
                if re.match(regex_pattern, value):
                    valid_values.append(value)
                else:
                    invalid_values.append(value)

    return valid_values, invalid_values

# 示例用法
csv_file = 'data.csv'
column_index = 2
regex_pattern = r'^[A-Za-z]+$'  # 匹配只包含字母的值

valid_values, invalid_values = check_valid_values(csv_file, column_index, regex_pattern)

print('Valid values:', valid_values)
print('Invalid values:', invalid_values)