python-3.x - 在 Pandas 数据框行中保留唯一的单词

Question

数据框:

> df
>type(df)
pandas.core.frame.DataFrame

ID      Property Type                                Amenities
1952043 Apartment, Villa, Apartment                  Park, Jogging Track, Park
1918916 Bungalow, Cottage House, Cottage, Bungalow   Garden, Play Ground

如何在数据框行中只保留由“逗号”分隔的唯一词？在这种情况下，它不得将“Cottage House”和“Cottage”视为相同。它必须检查数据框的所有列。所以我想要的输出应该如下所示:期望的输出:

    ID      Property Type                      Amenities
    1952043 Apartment, Villa                   Park, Jogging Track
    1918916 Bungalow, Cottage House, Cottage   Garden, Play Ground

Answer 1

首先，我创建了一个函数来为给定的字符串执行您想要的操作。其次，我将此函数应用于列中的所有字符串。

import numpy as np
import pandas as pd

df = pd.DataFrame([['Apartment, Villa, Apartment',
                    'Park, Jogging Track, Park'],
                   ['Bungalow, Cottage House, Cottage, Bungalow',
                    'Garden, Play Ground']],
                  columns=['Property Type', 'Amenities'])

def drop_duplicates(row):
    # Split string by ', ', drop duplicates and join back.
    words = row.split(', ')
    return ', '.join(np.unique(words).tolist())

# drop_duplicates is applied to all rows of df.
df['Property Type'] = df['Property Type'].apply(drop_duplicates)
df['Amenities'] = df['Amenities'].apply(drop_duplicates)
print(df)