python - 如何在 Pandas 中获取多个值

Question

我想将一组文件名拆分为名称和文件类型。

但是，我得到如下错误。

正确的做法是什么？

谢谢。

数据框是:

             full_name
0            ocean.jpg
1            mountain.png
2            forest.gif
3            river.mp4
4            animal.avi
5            girl.pdf

df[['name', 'type']] = df['full_name'].str.split('.')
KeyError: "['name' 'type'] not in index"

Answer 1

选项 1
split 与 tolist 如果您只想要结果

pd.DataFrame(df.full_name.str.split('.').tolist(), columns = ['name', 'type'])

       name type
0     ocean  jpg
1  mountain  png
2    forest  gif
3     river  mp4
4    animal  avi
5      girl  pdf

选项 2(您非常接近这个选项)
split 与 expand=True 如果你想编辑原始数据框

df[['name', 'type']] = df['full_name'].str.split('.', expand=True)

      full_name      name type
0     ocean.jpg     ocean  jpg
1  mountain.png  mountain  png
2    forest.gif    forest  gif
3     river.mp4     river  mp4
4    animal.avi    animal  avi
5      girl.pdf      girl  pdf