如何从列类型列表中删除 pandas DataFrame 中的空值

Question

只是期待一种解决方案，从具有列表值的列中删除空值，在某种意义上，我们已经预先替换了一些字符串，它是列表的字符串表示列。

在df.color 中，我们只是将*._Blue 替换为空字符串：

示例数据框：

df = pd.DataFrame({ 'Bird': ["parrot", "Eagle", "Seagull"], 'color': [ "['Light_Blue','Green','Dark_Blue']", "['Sky_Blue','Black','White', 'Yellow','Gray']", "['White','Jet_Blue','Pink', 'Tan','Brown', 'Purple']"] })

>>> df
      Bird                                              color
0   parrot                 ['Light_Blue','Green','Dark_Blue']
1    Eagle      ['Sky_Blue','Black','White', 'Yellow','Gray']
2  Seagull  ['White','Jet_Blue','Pink', 'Tan','Brown', 'Pu...

上述DF的结果：

>>> df['color'].str.replace(r'\w+_Blue\b', '')
0                                 ['','Green','']
1           ['','Black','White', 'Yellow','Gray']
2    ['White','','Pink', 'Tan','Brown', 'Purple']
Name: color, dtype: object

通常在python中很容易做到如下......

>>> lst = ['','Green','']
>>> [x for x in lst if x]
['Green']

恐怕可以做到以下几点。

df.color.mask(df == ' ')

Answer 1

使用filter和apply的另一种方式：

(df['color'].str.replace(r'\w+_Blue\b', '')
     .apply(lambda x: list(filter(bool, ast.literal_eval(x)))))

---

0                              [Green]
1         [Black, White, Yellow, Gray]
2    [White, Pink, Tan, Brown, Purple]

Answer 2

您没有一列列表，而是有一列包含列表的字符串表示形式。您可以使用ast.literal_eval 和str.endswith 一步完成所有操作。我会在这里使用列表理解，它应该比 apply 更快

---

import ast

fixed = [
    [el for el in lst if not el.endswith("Blue")]
    for lst in df['color'].apply(ast.literal_eval)
]

df.assign(color=fixed)

      Bird                              color
0   parrot                            [Green]
1    Eagle       [Black, White, Yellow, Gray]
2  Seagull  [White, Pink, Tan, Brown, Purple]

Answer 3

您可以使用explode(pandas 0.25.0) 然后将列表连接回来

 df['color'].str.replace(r'\w+_Blue\b', '').explode().loc[lambda x : x!=''].groupby(level=0).apply(list)