根据列值在 pandas dataframe 中选择行时出错

Question

我有一个 dataframe df ，它看起来像这样：

col1  col2 col3
 A     45    4
 A     3     5
 B     2     5

我想制作一个单独的 dataframe, df2 ，它只有df中的col等于A的行。 因此它应该看起来像：

col1  col2 col3
 A     45    4
 A     3     5

我只是使用df2=df1.loc[df1['col1']=='A'] 。 但是，这会返回错误： ValueError: Cannot index with multidimensional key 。 知道这里出了什么问题吗？

Answer 1

你尝试的对我有用，你可以试试这个：

df2 = df[df.col1 == 'A']

Output

  col1  col2    col3
0   A   45      4
1   A   3       5

编辑在 pandas 版本上测试

pd.__version__
'1.2.4'

Answer 2

试试看嘛：

df.query("col1=='A'")