Question

我有一个Pandas数据框，其中每一行都是一个列表。

我想搜索一个值，但是出现错误。而且我知道我的价值存在。

我检查一下：

df["text list"][1] == ['رهبری']

并得到：

True

那么我需要这个：

df[df["text list"] == ['رهبری']]

并收到此错误：

    ValueError                                Traceback (most recent call last)
    <ipython-input-42-f14f1b2306ec> in <module>
    ----> 1 df[df["text list"] == ['رهبری']]

    ~/.local/lib/python3.6/site-packages/pandas/core/ops/__init__.py in wrapper(self, other, axis)
       1205             # as it will broadcast
       1206             if other.ndim != 0 and len(self) != len(other):
    -> 1207                 raise ValueError("Lengths must match to compare")
       1208 
       1209             res_values = na_op(self.values, np.asarray(other))

    ValueError: Lengths must match to compare

Answer 1

当您将列表直接传递到DataFrame进行比较时，它期望一个大小相同的数组进行元素明智的比较。

为避免这种情况，我们可以使用apply来检查列表中是否存在每一行：

# example dataframe
>>> df = pd.DataFrame({'text list':[['aaa'], ['bbb'], ['ccc']]})
>>> df
  text list
0     [aaa]
1     [bbb]
2     [ccc]

使用Series.apply检查[bbb]：

>>> m = df['text list'].apply(lambda x: x == ['bbb'])
>>> df[m]
  text list
1     [bbb]

由于我们使用的是apply，它实际上是后台的“循环”实现。我们可以避免使用熊猫的开销并使用列表理解：

>>> m = [x == ['bbb'] for x in df['text list']]
>>> df[m]
  text list
1     [bbb]

列表的熊猫数据框的python问题

1 个答案: