从pandas数据框中删除带有空列表的行

Question

我有一个数据框，其中一些列包含空列表，另一些列包含字符串列表：

       donation_orgs                              donation_context
0            []                                           []
1   [the research of Dr. ...]   [In lieu of flowers , memorial donations ...]

我尝试返回的数据集中没有任何有空列表的行。

我试过检查空值：

dfnotnull = df[df.donation_orgs != []]
dfnotnull

和

dfnotnull = df[df.notnull().any(axis=1)]
pd.options.display.max_rows=500
dfnotnull

我已经尝试循环并检查存在的值，但我认为列表不会像我认为的那样返回Null或None：

dfnotnull = pd.DataFrame(columns=('donation_orgs', 'donation_context'))
for i in range(0,len(df)):
    if df['donation_orgs'].iloc(i):
        dfnotnull.loc[i] = df.iloc[i]

上述三种方法都只返回原始数据框中的每一行。=

Answer 1

您可以尝试切片，就像数据框是字符串而不是列表一样：

import pandas as pd
df = pd.DataFrame({
'donation_orgs' : [[], ['the research of Dr.']],
'donation_context': [[], ['In lieu of flowers , memorial donations']]})

df[df.astype(str)['donation_orgs'] != '[]']

Out[9]: 
                            donation_context          donation_orgs
1  [In lieu of flowers , memorial donations]  [the research of Dr.]

Answer 2

为避免转换为str并实际使用list，您可以这样做：

df[df['donation_orgs'].map(lambda d: len(d)) > 0]

它将donation_orgs列映射到每行列的长度，并仅保留至少一个元素的，过滤掉空列表。

返回

Out[1]: 
                            donation_context          donation_orgs
1  [In lieu of flowers , memorial donations]  [the research of Dr.]

正如所料。

Answer 3

假设您从CSV读取数据，另一种可能的解决方案可能是：

import pandas as pd

df = pd.read_csv('data.csv', na_filter=True, na_values='[]')
df.dropna()

na_filter定义了附加字符串，以识别为NaN。我在pandas-0.24.2上进行了测试。

Answer 4

您可以使用以下单行：

df[(df['donation_orgs'].str.len() != 0) | (df['donation_context'].str.len() != 0)]