如何使用列包含行且需要拆分的列来操作pandas DataFrame

Question

我有一个类似于表A的熊猫数据框，我想获取表B。用熊猫做这件事的最简单方法是什么？

谢谢

表A（ColofInt具有不同长度的字符串要解析）：

ColA ColB ColofInt             ColD 
A     B   StrA;StrB;StrC;       1
A     B   StrD;StrB;StrC;StrD;  3
A     B   StrC;StrB;            2
A     B   StrB;                 5

表B：

ColA ColB ColofInt1     ColofInt2 ColofInt2 ColofInt3  ColD 
A     B   StrA            StrB      StrC                1
A     B   StrD            StrB      StrC    StrD        3
A     B   StrC            StrB                          2
A     B   StrB                                          5

Answer 1

假设文件“ tableA.csv”包含以下内容：

ColA,ColB,ColofInt,ColD 
A,B,StrA;StrB;StrC;,1
A,B,StrD;StrB;StrC;StrD;,3
A,B,StrC;StrB;,2
A,B,StrB;,5

然后：

import pandas as pd
tableA= pd.read_csv('tableA.csv')

这将使用您的新列生成一个数据框

data_aux = pd.DataFrame(list(tableA.ColofInt.str.split(';').apply(lambda x: x[:-1])))
cols = []
for e in data_aux .columns:
    cols.append('ColofInt' + str(e+1)) 
data_aux .columns = cols

继承人“ data_aux”：

   ColofInt1    ColofInt2   ColofInt3   ColofInt4
0   StrA        StrB        StrC        None
1   StrD        StrB        StrC        StrD
2   StrC        StrB        None        None
3   StrB        None        None        None

并且这将连接数据框，并删除原始列。

tableB = pd.concat([tableA,data_aux],axis=1).drop('ColofInt',axis=1)

这是生成的“ tableB”：

   ColA ColB    ColD    ColofInt1   ColofInt2   ColofInt3   ColofInt4
0   A   B       1       StrA        StrB        StrC        None
1   A   B       3       StrD        StrB        StrC        StrD
2   A   B       2       StrC        StrB        None        None
3   A   B       5       StrB        None        None        None