Python如何将一个数据帧的两列组合成一个列表?

Mainland • 5 年前 • 1539 次点击

df = 

index    data1    data2
0         20       120
1         30       456
2         40       34

如何将上面df中的两列组合成一个列表,以便第一行元素先到第二行。

my_list = [20,120,30,456,40,34]

我的代码:

list1 = df['data1'].tolist()
list2 = df['data2'].tolist()

my_list = list1+list2

这不管用吗?

Python社区是高质量的Python/Django开发社区
本文地址：http://www.python88.com/topic/53882

1539 次点击

文章 [ 2 ] | 最新文章 5 年前

• 1 楼

ALollz 5 年前

array([[row1], [row2], ..., [rowN]]) 所以我们可以 ravel 应该很快。

df[['data1', 'data2']].to_numpy().ravel().tolist()
#[20, 120, 30, 456, 40, 34]

import perfplot
import pandas as pd
import numpy as np
from itertools import chain

perfplot.show(
    setup=lambda n: pd.DataFrame(np.random.randint(1, 10, (n, 2))), 
    kernels=[
        lambda df: df[[0, 1]].to_numpy().ravel().tolist(),
        lambda df: [x for i in zip(df[0], df[1]) for x in i],
        lambda df: [*chain.from_iterable(df[[0,1]].to_numpy())],
        lambda df: df[[0,1]].stack().tolist()  #  proposed by @anky_91
    ],
    labels=['ravel', 'zip', 'chain', 'stack'],
    n_range=[2 ** k for k in range(20)],
    equality_check=np.allclose,  
    xlabel="len(df)"
)

• 2 楼

U10-Forward 5 年前

这不起作用,因为它不会按同一索引添加,请使用以下列表理解:

print([x for i in zip(df['data1'], df['data2']) for x in i])

登录后回复