將包含字符串內的列表的數據框列拆分為兩列

Question

我有一個df，其中一列，'vals'包含一個字符串內的列表。 我想把它轉換成兩列'val1'和'vals2。 我試圖拆分和剝離字符串，但無法找到實現為df中的每一行執行此操作。

    vals
'[12.1, 15.0]'

val1  val2
12.1  15.0

Answer 1

如有必要，使用strip split和轉換的strip來浮動，最后通過add_prefix添加前綴：

df = pd.DataFrame({'vals':["'[12.1, 15.0]'","'[12.1, 15.0]'"]})

df = (df['vals'].str.strip("'[]")
               .str.split(', ', expand=True)
               .astype(float)
               .add_prefix('val'))

如果沒有缺失值和性能很重要：

df =  pd.DataFrame([x.strip("'[]").split(', ') for x in df['vals']], 
                    columns = ['val1', 'val2']).astype(float)

Answer 2

在這里我們使用ast.literal_eval -

import ast
df = pd.DataFrame({'A':['[12.1, 15.0]','[12.4, 11.1]']})
df['val1']=df['A'].apply(lambda x:ast.literal_eval(x)[0])
df['val2']=df['A'].apply(lambda x:ast.literal_eval(x)[1])
df    
                A  val1  val2
  0  [12.1, 15.0]  12.1  15.0
  1  [12.4, 11.1]  12.4  11.1