Как отформатировать фрейм данных в Python с несколькими столбцами, но одной строкой?

Question

У меня есть датафрейм в Python, который состоит из 1 строки, но 100 столбцов. Это выглядит так:

_id  d.0.id  d.0.name  d.0.dep.id  d.0.dep.name  d.0.dep.1.id  d.0.dep.1.name .... 

 A     B        C          D            E            F            G

Мне нужно преобразовать фрейм данных в CSV-файл следующим образом:

_id  d.0.id  d.0.name  d.dep.id   d.dep.name

 A     B        C          D           E
 A     B        C          F           G
 .     .        .          .           .
 .     .        .          .           .

Новый CSV-файл должен иметь 5 столбцов. Первые три столбца должны содержать одинаковые записи по всему файлу. Только последние два столбца изменяются в каждой строке. Пожалуйста, скажите мне, как я могу выполнять этот тип манипуляции с данными в Python, используя Pandas.

Answer 1

Получите новый фрейм данных с этими 5 столбцами:

df1 = df.iloc[:,0:5]
or
df1=df[[_id,d.0.id,d.0.name,d.dep.id,d.dep.name]]

Сохранить новый фрейм данных как csv:

df1.to_csv('./file_path')

Решение расширенного вопроса

Преобразовать одну строку в ndarray

import pandas as pd
df = pd.read_csv('df_info.txt', sep=",", header=0)
vallist=df.as_matrix()[0]

выход:

array(['NBA', 'Mens', 'Sports', 'LAL', 'Lakers', 'BOS', 'Celtics', 'SAS',
       'Spurs', 'OKC', 'Thunder'], dtype=object)

Создать Dict для хранения значений

dict={}
n=4
varlist1=[]
for i in range(len(vallist)):
    if(n<=9):
        dict[i]={}
        print(vallist[n],vallist[n+1])
        dict[i]['col1']=vallist[0]
        dict[i]['col2']=vallist[1]
        dict[i]['col3']=vallist[2]
        dict[i]['col4']=vallist[n]
        dict[i]['col5']=vallist[n+1]
        n+=2

Импортировать данные в Dataframe

df2=pd.DataFrame.from_dict(dict)
df2.transpose()

Окончательный результат:

   col1 col2    col3    col4    col5
0   NBA Mens    Sports  Lakers  BOS
1   NBA Mens    Sports  Celtics SAS
2   NBA Mens    Sports  Spurs   OKC