Добавить список из столбца панд в список Python - PullRequest
0 голосов
/ 10 мая 2018

У меня есть значения в списке в столбце панд, например:

df

id       col1
1       [51.97559, 4.12565]
2       [52.97559, 3.12565]
3       [49.97559, 5.12565]

Но когда я добавляю col1 в список, я получаю цитату вокруг первого элемента в каждом подсписке.

new_list = []
for val in df['col1'].values:
    new_list.append(val)

И я получил:

[['51.97559', 4.12565]
['52.97559', 3.12565]
['49.97559', 5.12565]]

Но мне нужно:

[[51.97559, 4.12565]
[52.97559, 3.12565]
[49.97559, 5.12565]]

Ответы [ 2 ]

0 голосов
/ 10 мая 2018

Угробить петлю. Давайте векторизовать это.

new_list = np.array(df['col1'].tolist(), dtype=float).tolist()

Ответственность ; это не будет работать, если ваши списки не все одного размера. В этом случае вам потребуется итерация , но вам не понадобится map (то есть, если вы хотите повысить производительность):

new_list = [
    [float(j) for j in i] for i in df['col1'].tolist()
] 
0 голосов
/ 10 мая 2018

Вы можете привести все элементы списка в float

new_list = []
for val in df['col1'].values:
    new_list.append(list(map(float, val)))
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...