如何计算csv列中的元素数量，并根据该计数返回另一列中的元素？

import csv

# 读取CSV文件
def read_csv_file(file_path):
    data = []
    with open(file_path, 'r') as csv_file:
        reader = csv.reader(csv_file)
        for row in reader:
            data.append(row)
    return data

# 计算元素数量并返回另一列中的元素
def calculate_element_count(data, count_column_index, return_column_index, threshold):
    element_count = {}
    result = []
    for row in data:
        count_element = row[count_column_index]
        return_element = row[return_column_index]
        
        # 计数
        if count_element in element_count:
            element_count[count_element] += 1
        else:
            element_count[count_element] = 1
        
        # 返回另一列中的元素
        if element_count[count_element] > threshold:
            result.append(return_element)
    
    return result

# 示例用法
csv_file_path = 'data.csv'
count_column_index = 0  # 需要计数的列的索引
return_column_index = 1  # 需要返回的列的索引
threshold = 3  # 阈值，超过该计数的元素将返回

data = read_csv_file(csv_file_path)
result = calculate_element_count(data, count_column_index, return_column_index, threshold)
print(result)