错误：ValueError:在尝试筛选数据帧时，长度必须匹配才能进行比较

Question

我很抱歉问这个问题，但出于某些原因，我花了几分钟来弄清楚这一点，但我不明白

我有一个dataframe，这种性质的东西

df

Output:
  TypePro
  ["JJ", "KK"]
  ["JK", "RJ"]
  ["JK"]
  ["JK"]    

我正在尝试筛选dataframe：

df_JJ_KK = df[df.TypePro == ["JJ", "KK"]]

但是我得到了以下错误：ValueError: Lengths must match to compare

请问我如何解决这个问题

Answer 1

使用列表理解进行筛选：

df = df[[x == ["JJ", "KK"] for x in df.TypePro]]
print (df)
    TypePro
0  [JJ, KK]

或比较tuples：

df = df[df.TypePro.map(tuple) == tuple(["JJ", "KK"])]