Question

我有一个pandas DataFrame，其列在列中具有以下值：

Identifier
[1;12;7;3;0]
[4;5;2;6;0]

我想将此列中方括号中的值转换为5个新列。基本上，我想将这些值拆分为5个新列，同时保持新列的索引与原始列相同。

Identifier,a,b,c,d,e
[1;12;7;3;0],1,12,7,3,0
[4;5;2;6;0],4,5,2,6,0

pattern = re.compile(r'(\d+)')
for g in raw_data["Identifier"]:
    new_id = raw_data.Identifier.str.findall(pattern) # this converts the Identifier into a list of the 5 values
raw_data.append({'a':new_id[0],'b':new_id[1],'c':new_id[2],'d':new_id[3],'d':new_id[4]}, ignore_index=True)

上面的代码将提取的值从列“identifier”添加到DataFrame的末尾，而不是添加到相应的行。如何将提取的值添加到与原始列相同的行/索引（“标识符”）？

Answer 1

一种方法是使用str方法获取数字，从中创建新的数据帧，然后加入（或连接）结果。例如，

id_data = df.Identifier.str.strip("[]").str.split(";").tolist()
df_id = pd.DataFrame(id_data, columns=list("abcde"), index=df.index, dtype=int)
df2 = df.join(df_id)

产生类似

的东西

      Identifier  a   b  c  d  e
10  [1;12;7;3;0]  1  12  7  3  0
20   [4;5;2;6;0]  4   5  2  6  0

将pandas DataFrame中的列转换为多列

1 个答案: