python - 字符串在 Pandas 数据框中包含python中的函数？

Question

我是 python 的新手，所以这里可能有一个简单的解决方案。我正在尝试清理有关 Pandas 数据框中租金价格/平方英尺的数据集。我的卧室数据列包括有关卧室和平方英尺的信息。大多数条目的格式类似于“/1br - 950ft²”，但有些是“/1br”，有些是“/950ft²”。我正在尝试创建一个只有卧室的干净列，但由于格式问题，我不能只在某个字符后拆分字符串。

我决定我需要创建一个函数来测试字符串是否包含“br”，但我收到了一个错误。

这是我的代码:

def cleaned_bedrooms(x):
    if df[df['bedrooms'].str.contains('br')]:
        df['bedrooms'] = df['bedrooms'].str.split('-').str[0]
    else:
        return None
df['bedrooms'].map(cleaned_bedrooms)

我似乎已经设置了一个 bool 函数(我假设由 if 语句触发)，因为我得到的错误是“ValueError:DataFrame 的真值不明确。使用 a.empty，a。 bool()、a.item()、a.any() 或 a.all()。”对于包含 .map(cleaned_bedrooms)

的行

Answer 1

如果这是你的数据框，

    bedrooms
0   / 1br - 950ft²
1   / 1br
2   /950ft²

您可以使用 str.extract 提取卧室

df['bedrooms'] = df['bedrooms'].str.extract('(\d+?br)', expand = False)

你得到

    bedrooms
0   1br
1   1br
2   NaN