迭代 pandas 列值

Question

我有一个数据集，其中有一列字符串变量 (~6000) 描述了浴室类型（即 1 个私人浴室、2 个共用浴室等），有没有办法让我“查找并替换”这些变量以简化为简单的数字， 1 个私人浴室更换为 integer 1，依此类推。

目前我尝试使用 the.replace function 是一种冗长且低效的方法。

db["bathrooms_text"]= db["bathrooms_text"].replace("0 baths", 0)

db["bathrooms_text"]= db["bathrooms_text"].replace("1.5 baths", 1.5)

db["bathrooms_text"]= db["bathrooms_text"].replace("2 baths", 2)

Answer 1

使用str.extract ：

db['bathrooms_text'] = db['bathrooms_text'].str.extract('(\d+\.?\d*)')
print(db)

# Output
  bathrooms_text
0              0
1            1.5
2              2