Как сделать первые три элемента в списке как столбцы, а остальные как строки - Python - PullRequest
Новая
       37

Как сделать первые три элемента в списке как столбцы, а остальные как строки - Python

2 голосов
/ 09 июля 2019 
df = pd.DataFrame({'Categotry':['Food','Animal'],
          'Detail':[['Name','Color','Sweet?','Bread','Brown','No','Rice','White','No','Sushi','N/A','No'],
                    ['Name','Predator?','Habitat','Tigers','Yes','Forests','Lions','Yes','Savanna','Deers','No','Hardwoods']]})

У меня есть вышеуказанный фрейм данных, и я хочу разделить столбец Сведения, как показано ниже: enter image description here

Как это сделать в Python?

Спасибо за помощь.

1 Ответ

2 голосов
/ 09 июля 2019

Пользовательская функция 

def process_details(details):
    cols, *data = np.reshape(details, (-1, 3))
    return pd.DataFrame(data, columns=cols)

Я использую np.reshape, потому что я к этому привык. Однако это может сделать то же самое. 

def process_details(details):
    cols, *data = zip(*[iter(details)] * 3)
    return pd.DataFrame(data, columns=cols)

Бок о бок

Поскольку имена столбцов не совпадают 

pd.concat({
    cat: process_details(details)
    for cat, details in zip(*map(df.get, df))
}, sort=False, axis=1)

   Animal                        Food              
     Name Predator?    Habitat   Name  Color Sweet?
0  Tigers       Yes    Forests  Bread  Brown     No
1   Lions       Yes    Savanna   Rice  White     No
2   Deers        No  Hardwoods  Sushi    N/A     No

Но если вы настаиваете на их укладке 

pd.concat({
    cat: process_details(details)
    for cat, details in zip(*map(df.get, df))
}, sort=False)

            Name Predator?    Habitat  Color Sweet?
Animal 0  Tigers       Yes    Forests    NaN    NaN
       1   Lions       Yes    Savanna    NaN    NaN
       2   Deers        No  Hardwoods    NaN    NaN
Food   0   Bread       NaN        NaN  Brown     No
       1    Rice       NaN        NaN  White     No
       2   Sushi       NaN        NaN    N/A     No
...