读取带有间距和逗号的多个列表的文本文件，列表中的元素之间存在熊猫数据框

Question

我有一个称为tropical.txt的文本文件，该文件具有多个列表，每个列表均由新行分开。请注意，逗号被空间包围。

 space here and space here
         | |
['papaya' , 'mangosteen' , 'banana']
[]
['coconut' , 'mango']
['mangosteen' , 'papaya']

我尝试了以下代码

import pandas as pd

df = pd.read_csv('tropical.txt', sep='\n', header=None, engine = 'python')
df

，

ValueError: Specified \n as separator or delimiter. This forces the python engine which does not accept a line terminator. Hence it is not allowed to use the line terminator as separator.

如果我要做

import pandas as pd

df = pd.read_csv('tropical.txt', header= None, engine = 'python')
df

输出，我期望

         0           1             2
0   ['papaya'   'mangosteen'    'banana']
1   []               None        None
2   ['coconut'      'mango']     None
3   ['mangosteen'   'papaya']    None

有

                        0   
0   [papaya,mangosteen,banana]
1   []  
2   [coconut,mango] 
3   [mangosteen,papaya]

任何建议吗？

Answer 1

您可以通过指定不会在行中出现的分离器（例如\ 0）来使用read_csv（以便整体读取每行）和 aSt.literal_eval 作为转换器的转换器值：

import ast

pd.read_csv('tropical.txt', header=None, sep='\0', names=['fruits'], converters={ 'fruits' : ast.literal_eval })

输出：

                         fruits
0  [papaya, mangosteen, banana]
1                            []
2              [coconut, mango]
3          [mangosteen, papaya]