python - Pandas 爆炸功能不适用于字符串列列表

Question

要像列到行一样分解列表，我们可以使用 pandas explode()功能。我的 Pandas 版 ' 0.25.3 '
给定的 example为我工作，Stackoverflow.com 的另一个答案按预期工作，但不适用于我的数据集。

    city        nested_city
0   soto        ['Soto']
1   tera-kora   ['Daniel']
2   jan-thiel   ['Jan Thiel']
3   westpunt    ['Westpunt']
4   nieuwpoort  ['Nieuwpoort', 'Santa Barbara Plantation']

我试过的:

test_data['nested_city'].explode()

和

test_data.set_index(['nested_city']).apply(pd.Series.explode).reset_index()

输出

0    ['Soto']                                  
1    ['Daniel']                                
2    ['Jan Thiel']                             
3    ['Westpunt']                              
4    ['Nieuwpoort', 'Santa Barbara Plantation']
Name: neighbors, dtype: object

Answer 1

您需要确保您的列是列表类型才能使用 Pandas 的 explode() .这是一个有效的解决方案:

from ast import literal_eval

test_data['nested_city'] = test_data['nested_city'].apply(literal_eval) #convert to list type
test_data['nested_city'].explode()

要一次分解多个列，您可以执行以下操作:

not_list_cols = [col for col in test_data.columns if col not in ['col1', 'col2']] #list of columns you are not exploding (assume col1 and col2 are being exploded)
test_data = test_data.set_index(not_list_cols).apply(pd.Series.explode).reset_index()