如何将列表中的前三个元素设置为列，其余元素设置为行

Question

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']]})

我有上面的dataframe，我想将Detail列拆分如下：

在Python中如何做到这一点？你知道吗

谢谢你的帮助。你知道吗

Answer 1

自定义函数

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