python - 获取列表的列表，拆分列表，然后转换为数据帧

Question

我正在尝试获取如下所示的列表列表(我们称之为“全部”):

[['Albany County, Wyoming', '56', '001'],
 ['Big Horn County, Wyoming', '56', '003'],
 ['Campbell County, Wyoming', '56', '005'],...

并将其转换为如下所示的数据框:

COUNTY            STATE     FIPS1 FIPS2
'Albany County'   'Wyoming'  '56'  '001'
'Big Horn County' 'Wyoming'  '56'  '003'
'Campbell County' 'Wyoming'  '56'  '005'

这是我迄今为止的列表列表:

for index, line in enumerate(All):
    All[index] = line[0].split(','), line[1:]

pd.DataFrame(All)

但这正在返回:

                            0          1
0        [Albany County,  Wyoming]  [56, 001]
1      [Big Horn County,  Wyoming]  [56, 003]
2      [Campbell County,  Wyoming]  [56, 005]

预先感谢您的帮助!

Answer 1

All = [
    ['Albany County, Wyoming', '56', '001'],
    ['Big Horn County, Wyoming', '56', '003'],
    ['Campbell County, Wyoming', '56', '005']
]

pd.DataFrame(
    [line[0].split(', ') + line[1:] for line in All],
    columns=['COUNTY', 'STATE', 'FIPS1', 'FIPS2']
)

            COUNTY    STATE FIPS1 FIPS2
0    Albany County  Wyoming    56   001
1  Big Horn County  Wyoming    56   003
2  Campbell County  Wyoming    56   005

<小时/>

或者

df = pd.DataFrame(All, columns=['COUNTY, STATE', 'FIPS1', 'FIPS2'])

col = 'COUNTY, STATE'
df[col].str.extract(
    '(?P<COUNTY>.*), (?P<STATE>.*)', expand=True
).join(df.drop(col, 1))

            COUNTY    STATE FIPS1 FIPS2
0    Albany County  Wyoming    56   001
1  Big Horn County  Wyoming    56   003
2  Campbell County  Wyoming    56   005