如何将 Pandas Dataframe 列转换为“列表”类型？

Question

遵循Pandas expand()方法文档，可以将一行中的列表分解为多行：

df = pd.DataFrame({'A': [[1, 2, 3], 'foo', None, [3, 4]], 'B': 1})
df.explode('A')

但是，我从数据库中获取的 DataFrame() 包含被视为字符串的列表：

df = pd.DataFrame({'A': ["[1, 2, 3]", 'foo', None, "[3, 4]"], 'B': 1})
df.explode('A')
# Does not fail, but does not explode "[1, 2, 3]"

在 Pandas 之外，我使用ast.literal_eval()但不知道如何使它适用于我的专栏。

如何将我的 'A' 列转换为列表，以便 expand('A') 工作？

Answer 1

您可以尝试使用DataFrame.apply ：

def f(x):
    try:
        return ast.literal_eval(x)
    except Exception:
        return x


df['A']=df['A'].apply(f)
df.explode('A')

      A  B
0     1  1
0     2  1
0     3  1
1   foo  1
2  None  1
3     3  1
3     4  1