Разделить столбец данных, содержащий список внутри строки, на два столбца - PullRequest
0 голосов
/ 09 января 2019

У меня есть df, где в одном столбце 'vals' содержится список внутри строки. Я хочу преобразовать это в две колонки 'val1' и 'vals2. Я попытался разделить и удалить строку, но не могу найти реализацию, которая бы делала это для каждой строки в df.

    vals
'[12.1, 15.0]'

val1  val2
12.1  15.0

Ответы [ 2 ]

0 голосов
/ 09 января 2019

Здесь мы используем 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
0 голосов
/ 09 января 2019

Используйте strip с split и, если необходимо, приведите к плавающим числам, добавьте префикс последним на 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)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...