我有一个数据框,其中一些列包含空列表,其他列包含字符串列表:
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]
以上三种方法都只是返回原始数据框中的每一行。=
为了避免转换为
str
并实际使用 list
,您可以这样做:
df[df['donation_orgs'].map(len) > 0]
它将
donation_orgs
列映射到每行的列表的长度,并仅保留具有至少一个元素的列表,过滤掉空列表。
它回来了
Out[1]:
donation_context donation_orgs
1 [In lieu of flowers , memorial donations] [the research of Dr.]
正如预期的那样。
您可以尝试切片,就好像数据框是字符串而不是列表一样:
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.]
您可以使用以下一行:
df[(df['donation_orgs'].str.len() != 0) | (df['donation_context'].str.len() != 0)]
假设您从 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
上测试了这个。
可能是数据类型不同,这可能会有帮助
df[df.astype(str)['donation_orgs'] != '[]']