编写返回dataset中某列中唯一名称数量的函数- Python

import pandas as pd

def count_unique_names(dataset, column_name):
    """
    计算数据集中某列的唯一名称数量。

    参数:
    dataset (pd.DataFrame): 包含数据的Pandas DataFrame。
    column_name (str): 需要计算唯一值的列名。

    返回:
    int: 列中唯一名称的数量。
    """
    # 使用nunique()函数计算唯一值的数量
    unique_count = dataset[column_name].nunique()
    return unique_count

# 示例使用
if __name__ == "__main__":
    # 创建一个示例DataFrame
    data = {
        'Name': ['Alice', 'Bob', 'Alice', 'Charlie', 'Bob', 'David'],
        'Age': [24, 27, 24, 22, 27, 30]
    }
    df = pd.DataFrame(data)

    # 调用函数并打印结果
    column_name = 'Name'
    print(f"列 '{column_name}' 中唯一名称的数量是: {count_unique_names(df, column_name)}")

# 如果需要将NaN视作一个独特的值
unique_count_with_nan = dataset[column_name].fillna('missing').nunique()