Question

我有以下df，我想创建一个if语句以仅打印特定的行。

pandas_dataframe = pd.DataFrame({'actors' : [['Tom Hanks', 'Tim Allen', 'Don Rickles', 'Jim Varney'],
                                             ['Robin Williams', 'Jonathan Hyde', 'Kirsten Dunst'],
                                             ['Walter Matthau', 'Jack Lemmon', 'Sophia Loren', 'n', 'ix']],
                                 'movie':['Toy Story', 'Jumanji', 'X-men']})

我只想打印演员列表中有一个演员的长度为1的行，所以我只想打印第三行，因为它至少有1个演员的长度为1。

+--------------------------------------------------------------+------+
| actors                                                       |movie |
+--------------------------------------------------------------+------
|['Walter Matthau', 'Jack Lemmon', 'Sophia Loren', 'n', 'ix']  | X-men|
+--------------------------------------------------------------+------+

Answer 1

尝试：

import numpy as np

mask=pandas_dataframe.actors.explode().str.len().eq(1)
res=pandas_dataframe.loc[np.unique(mask.loc[mask].index)]

输出：

                                              actors  movie
2  [Walter Matthau, Jack Lemmon, Sophia Loren, n,...  X-men

Answer 2

使用lambda函数并应用：

import pandas as pd

df = pd.DataFrame(
    {
        "actors": [
            ["Tom Hanks", "Tim Allen", "Don Rickles", "Jim Varney"],
            ["Robin Williams", "Jonathan Hyde", "Kirsten Dunst"],
            ["Walter Matthau", "Jack Lemmon", "Sophia Loren", "n", "ix"],
        ],
        "movie": ["Toy Story", "Jumanji", "X-men"],
    }
)

filt = df.actors.apply(lambda x: any(len(y) == 1 for y in x))
df = df[filt]
print(df)

返回：

                                              actors   movie
2  [Walter Matthau, Jack Lemmon, Sophia Loren, n, ix]  X-men

Answer 3

您可以使用以下代码，在其中一行中过滤数据框。

df[df.actors.apply(lambda o: any([len(x)==1 for x in o]))]

根据应用于列表的if条件过滤数据框

3 个答案: