解析列以重塑数据框

Question

我有一个 csv，我使用 pandas 作为数据框导入。列如下：

Step1:A Step1:B Step1:C Step1:D Step2:A Step2:B Step2:D Step3:B Step3:D Step3:E
0         1        2        3       4      5       6       7       8     9

步骤和参数之间用“:”分隔。我想将数据框重塑为如下所示：

    Step1   Step2   Step3
A   0          4    nan
B   1          5      7
C   2        nan    nan
D   3          6      8
E   nan      nan      9

现在，如果我想维护列顺序，这样我就有了这种情况：

Step2:A  Step2:B    Step2:C  Step2:D    Step1:A   Step1:B   Step1:D  AStep3:B   AStep3:D    AStep3:E
0        1          2        3          4         5         6        7          8           9

步骤和参数之间用“:”分隔。我想将数据框重塑为如下所示：

    Step2   Step1   AStep3
A   0          4    nan
B   1          5      7
C   2        nan    nan
D   3          6      8
E   nan      nan      9

Answer 1

尝试 read_csv 和 delim_whitespace:

df = pd.read_csv('file.csv', delim_whitespace=True)
df.columns = df.columns.str.split(':', expand=True)
df.stack().reset_index(level=0, drop=True)

输出：

   Step1  Step2  Step3
A    0.0    4.0    NaN
B    1.0    5.0    7.0
C    2.0    NaN    NaN
D    3.0    6.0    8.0
E    NaN    NaN    9.0

解析列以重塑数据框

Parse columns to reshape dataframe

python

regex

csv

reshape

dataframe