将多个列表放入数据框


164

如何获取多个列表并将它们作为不同的列放在python数据框中?我尝试了此解决方案,但遇到了一些麻烦。

尝试1:

  • 有三个列表,并将它们压缩在一起并使用 res = zip(lst1,lst2,lst3)
  • 产量仅一栏

尝试2:

percentile_list = pd.DataFrame({'lst1Tite' : [lst1],
                                'lst2Tite' : [lst2],
                                'lst3Tite' : [lst3] }, 
                                columns=['lst1Tite','lst1Tite', 'lst1Tite'])
  • 产生一行3列(按上述方式),或者如果我转置则为3行1列

如何获得100行(每个独立列表的长度)乘3列(三个列表)的熊猫数据框?

Answers:


279

我认为您快到了,请尝试删除lsts 周围的多余方括号(此外,从像这样的字典创建数据框时,您无需指定列名):

import pandas as pd
lst1 = range(100)
lst2 = range(100)
lst3 = range(100)
percentile_list = pd.DataFrame(
    {'lst1Title': lst1,
     'lst2Title': lst2,
     'lst3Title': lst3
    })

percentile_list
    lst1Title  lst2Title  lst3Title
0          0         0         0
1          1         1         1
2          2         2         2
3          3         3         3
4          4         4         4
5          5         5         5
6          6         6         6
...

如果您需要一种性能更高的解决方案,np.column_stack而不是zip第一次尝试,可以使用此方法,此处示例的速度大约提高了2倍,但是我认为这会降低可读性:

import numpy as np
percentile_list = pd.DataFrame(np.column_stack([lst1, lst2, lst3]), 
                               columns=['lst1Title', 'lst2Title', 'lst3Title'])

是np.column_stack视图,还是复制数据。(如果是复制的话,似乎效率可能更高(O(1),而不是O(n))
。– user48956

@maxymoo列名称可以自动设置为列表名称吗?
joe5

1
如果列表具有不同的数据类型,则numpy列堆栈将无法正常工作
user6386155

54

在此处添加到Aditya Guru的答案中。无需使用地图。您可以通过以下方式简单地做到这一点:

pd.DataFrame(list(zip(lst1, lst2, lst3)))

这会将列的名称设置为0,1,2。要设置自己的列名,可以将关键字参数传递columns给上述方法。

pd.DataFrame(list(zip(lst1, lst2, lst3)),
              columns=['lst1_title','lst2_title', 'lst3_title'])

3
在Python 3.8和Pandas 1.0中,我们不需要使用列表函数,因为DataFrame期望可迭代,而zip()返回可迭代对象。所以,pd.DataFrame(zip(lst1, lst2, lst3))也应该这样做。
Sarfraaz Ahmed


8

添加了另一种可扩展的解决方案。

lists = [lst1, lst2, lst3, lst4]
df = pd.concat([pd.Series(x) for x in lists], axis=1)

你能解释一下吗?
ZakS

1
您垂直加入(concat)系列(axis = 1),以从列表中创建DataFrame
yona bentelac,

5

除了上述答案,我们可以随时创建

df= pd.DataFrame()
list1 = list(range(10))
list2 = list(range(10,20))
df['list1'] = list1
df['list2'] = list2
print(df)

希望能帮助到你 !


1

@oopsi已使用pd.concat(),但未包含列名称。您可以执行以下操作,与接受的答案中的第一个解决方案不同,该操作使您可以控制列顺序(避免使用无序的字典):

import pandas as pd
lst1 = range(100)
lst2 = range(100)
lst3 = range(100)

s1=pd.Series(lst1,name='lst1Title')
s2=pd.Series(lst2,name='lst2Title')
s3=pd.Series(lst3 ,name='lst3Title')
percentile_list = pd.concat([s1,s2,s3], axis=1)

percentile_list
Out[2]: 
    lst1Title  lst2Title  lst3Title
0           0          0          0
1           1          1          1
2           2          2          2
3           3          3          3
4           4          4          4
5           5          5          5
6           6          6          6
7           7          7          7
8           8          8          8
...

1

有多种方法可以从多个列表创建数据框。

list1=[1,2,3,4]
list2=[5,6,7,8]
list3=[9,10,11,12]
  1. pd.DataFrame({'list1':list1, 'list2':list2, 'list3'=list3})

  2. pd.DataFrame(data=zip(list1,list2,list3),columns=['list1','list2','list3'])


0

您可以简单地使用以下代码

train_data['labels']= train_data[["LABEL1","LABEL1","LABEL2","LABEL3","LABEL4","LABEL5","LABEL6","LABEL7"]].values.tolist()
train_df = pd.DataFrame(train_data, columns=['text','labels'])
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.