从列中具有重复值和非重复值的数据帧中创建列表

Question

我有这样的数据框架

data = [['Ma', 1,'too'], ['Ma', 1,'taa'], ['Ma', 1,'tuu',],['Ga', 2,'too'], ['Ga', 2,'taa'], ['Ga', 2,'tuu',]]
 

df = pd.DataFrame(data, columns = ['NAME', 'AID','SUBTYPE'])
 

NAME ID SUBTYPE
Ma    1     too
Ma    1     taa
Ma    1     tuu
Ga    2     too
Ga    2     taa
Ga    2     tuu

有重复的名称和ID以及不同的子类型

我想要一个这样的名单

Ma-1-[too,taa,too],Ga-2-[too,taa,tuu]

编辑:名称和ID应该总是相同的。

Answer 1

通常，为了在Python中实现这一点，我们将使用字典，因为键不能被复制。

# We combine the NAME and ID keys, so we can use them together as a key.
df["NAMEID"] = df["NAME"] + "-" + df["ID"].astype(str)

# Convert the desired fields to lists.
name_id_list = df["NAMEID"].tolist()
subtype_list = df["SUBTYPE"].tolist()

# Loop through the lists by zipping them together.
results_dict = {}
for name_id, subttype in zip(name_id_list, subtype_list):
    if results_dict.get(name_id):
        # If the key already exists then instead we append them to the end of the list. 
        results_dict[name_id].append(subttype)
    else:
        # If key not exists add them as key-value pairs to a dictionary.
        results_dict[name_id] = [subtype]

结果dict最终会看起来如下：

{'Ma-1': ['too', 'taa', 'tuu'], 'Ga-2': ['too', 'taa', 'tuu']}