Pandas：将JSON列表值拆分为新列

Question

我使用Pandas将CSV加载到以下DataFrame：

      value                                  values                                   
0        56.0             [-0.5554548,10.0748005,4.232949]                          
1        72.0         [-0.1953888,0.15093994,-0.058532715] 
...

现在我想用3个新列替换“values”列，如下所示：

     value     values_a      values_b      values_c                                      
0     56.0    -0.5554548    10.0748005     4.232949                          
1     72.0    -0.1953888    0.15093994    -0.058532715 
    ...

如何将列表拆分为3列？

Answer 1

你可以使用split删除[]的strip：

df1 = df.pop('values').str.strip('[]').str.split(',',expand=True).astype(float)
df[['values_a', 'values_b', 'values_c']] = df1

解决方案如果没有NaNs：

L = [x.split(',') for x in df.pop('values').str.strip('[]').values.tolist()]
df[['values_a', 'values_b', 'values_c']] = pd.DataFrame(L).astype(float)

首先将列转换为列表然后使用DataFrame构造函数的解决方案：

import ast
s = df.pop('values').apply(ast.literal_eval)
df[['values_a', 'values_b', 'values_c']] = pd.DataFrame(s.values.tolist()).astype(float)

类似：

df = pd.read_csv(file converters={'values':ast.literal_eval})
print (df)
   value                                  values
0   56.0      [-0.5554548, 10.0748005, 4.232949]
1   72.0  [-0.1953888, 0.15093994, -0.058532715]

df1 = pd.DataFrame(df.pop('values').tolist()).astype(float)
df[['values_a', 'values_b', 'values_c']] = df1

最后：

print (df)
   value  values_a   values_b  values_c
0   56.0 -0.555455  10.074801  4.232949
1   72.0 -0.195389   0.150940 -0.058533

编辑：

如果在某些列中可能更多为3值，则无法分配给3个新列。解决方案是使用join：

df = df.join(df1.add_prefix('val'))
print (df)
   value      val0       val1      val2
0   56.0 -0.555455  10.074801  4.232949
1   72.0 -0.195389   0.150940 -0.058533