您如何檢查您的 Class 是否具有具有唯一值的對象？

Question

我正在嘗試創建一個具有 user_id 屬性的 class 。 我想檢查所有用戶 ID 是否唯一。 id 最初來自 csv 文件。 使用 python，我需要導入並檢查用戶 ID 是否都是唯一的。 謝謝！

class Data:
    def __init__(self, user_id, title):
        self.user_id = user_id
        self.title = title
    

Answer 1

你可以做類似的事情

import pandas as pd
current_ids = pd.read_csv(path_to_your_csv_file)['user_id'].values
class Data:
    def __init__(self, user_id, title):
        self.user_id = user_id
        self.title = title
        if not self.check_for_uniqueness: raise ValueError("Non-unique user id")
        else: current_ids.append(self.user_id)

    def check_for_uniqueness(self):
        return self.user_id not in current_ids