python pandas loc返回空数据帧

Question

我正在尝试使用以下代码获取子集数据框：

dprev = "eiffel tower"

df.loc[df['place'] == dprev] -> returns empty

drandom = random.choice(df['place'].unique())

df.loc[df['place'] == drandom] -> returns the subset

为什么dprev是字符串变量时我没有看到同样的事情？

Answer 1

为此您不必使用df.loc

dprev = "eiffel tower"

df[df['place'] == dprev] 

See this answer

Answer 2

您可以尝试将str.contains与case=False

一起使用

<强>实施例

import pandas as pd

dprev = "eiffel tower"
df = pd.DataFrame({"place": ["eiffel tower", "Eiffel tower", "Hello"], "data":[1,2,3]})
print(df.loc[df['place'].str.contains(dprev, case=False)])

<强>输出：

   data         place
0     1  eiffel tower
1     2  Eiffel tower

Answer 3

尝试这种方式：

dprev = "eiffel tower"
df = df.loc[df['place'].str.lower() == dprev]

或：

df = df.loc[df['place'].str.lower().str.contains(dprev)]