CSV 中的关键字检测并添加到新列

Question

我在 CSV 文件（近 50 万个条目）中有房地产及其详细信息（17 列）。其中一列提供了一个位置，但实际上有点太详细了。我想对我的条目进行分类，因此我想简化位置以提供更多通用区域。我会将我想要将条目分类到列表中的区域，例如：

keywords = ['Downtown','Park View','Industrial District', ... ]

因此，理想情况下，我想获取一个具有例如 Sky Tower Downtown Los Angeles 的条目，然后将其归类为 Downtown。

所以任务是首先检测location 列中的关键字，然后将其附加到新列（如果可能，就在它旁边）。如果条目中没有找到关键字，我会将其归类为Other。

看起来像这样：

Date	Record_Type	Location	Proterty_Type	...	Price
19-Mar-21	Active Listing	Sky Tower Downtown Los Angeles	Apartment	...	15000
19-Mar-21	Active Listing	Central Park Residential Tower, 5th Avenue	Apartment	...	17000
20-Mar-21	Active Listing	Meadow Gardens, Park View	Villa	...	125000

类似于：

Date	Record_Type	Location	Area	Proterty_Type	...	Price
19-Mar-21	Active Listing	Sky Tower Downtown Los Angeles	Downtown	Apartment	...	15000
19-Mar-21	Active Listing	Central Park Residential Tower, 5th Avenue	Other	Apartment	...	17000
20-Mar-21	Active Listing	Meadow Gardens, Park View	Park View	Villa	...	125000

最后，它将所有内容保存到一个新的 csv 文件中。理想情况下，我还希望您使用pandas 在 csv 上读/写。

提前致谢！

编辑： 我尝试了以下线程之类的方法，但我得到了错误，我不知道出了什么问题，所以我愿意接受新的想法。

How to append a new column to a CSV file using Python?

Adding new column to CSV in Python

Answer 1

如果你有这个数据名：

        Date     Record_Type                                    Location Proterty_Type   Price
0  19-Mar-21  Active Listing              Sky Tower Downtown Los Angeles     Apartment   15000
1  19-Mar-21  Active Listing  Central Park Residential Tower, 5th Avenue     Apartment   17000
2  20-Mar-21  Active Listing                   Meadow Gardens, Park View         Villa  125000

然后：

keywords = ["Downtown", "Park View", "Industrial District"]

df.insert(
    loc=3,
    column="Area",
    value=df["Location"].apply(
        lambda x: next((kw for kw in keywords if kw in x), "Other")
    ),
)
print(df)

在Location 旁边创建Area 列并打印：

        Date     Record_Type                                    Location       Area Proterty_Type   Price
0  19-Mar-21  Active Listing              Sky Tower Downtown Los Angeles   Downtown     Apartment   15000
1  19-Mar-21  Active Listing  Central Park Residential Tower, 5th Avenue      Other     Apartment   17000
2  20-Mar-21  Active Listing                   Meadow Gardens, Park View  Park View         Villa  125000